1.收集數據
2.數據標注
pip install labelimg
3.劃分數據集 0.2的驗證機0.8的訓練集
import os
from shutil import copyfile
from sys import exit
import randomsource = r"D:\Data\imgs\screenc" + '\\'
target_train = r"D:\Data\imgs\datasets\mydata\images\train" + '\\'
target_valid = r"D:\Data\imgs\datasets\mydata\images\valid" + '\\'source_txt = r"D:\Data\imgs\lable" + '\\'
target_txt_train = r"D:\Data\imgs\datasets\mydata\lables\train" + '\\'
target_txt_valid = r"D:\Data\imgs\datasets\mydata\lables\valid" + '\\'def creat_dir(dirs):if not os.path.exists(dirs):os.makedirs(dirs)creat_dir(target_train)
creat_dir(target_valid)
creat_dir(target_txt_train)
creat_dir(target_txt_valid)# 顯示在所設置路徑下的所有圖片, filename這里僅為文件的文件名,如1.jpg
for filename in os.listdir(source):if '.png' in filename:source_filename = source + filename # 加一個根目錄編程圖片的路徑if(random.random()>0.2):target_train_filename = target_train + filename # 加一個根目錄編程圖片的路徑copyfile(source_filename, target_train_filename)source_txt_filename=source_txt+filename.split(".")[0]+".txt"copyfile(source_txt_filename, target_txt_train+filename.split(".")[0]+".txt")else:copyfile(source_filename, target_valid+ filename)copyfile(source_txt+filename.split(".")[0]+".txt", target_txt_valid+filename.split(".")[0]+".txt")
準確率(精確率),召回率
假定100人就診,12人真的生病,10人被醫生判為生病,9人被進一步檢查確認真的生病。
準確率:9/10=0.9
召回率 9/12=0.75
f1=92/(92+3+1)=0.81