在mac上用自己的数据训练YOLOv3的准备工作

1 准备数据集

1.1利用python爬虫收集图片

百度一下，用靠谱的爬虫方法，从网络上快速收集符合你要求的图片。

1.2数据增强

借助keras，对数据做旋转、平移等操作，以增大样本数量。

2 图片预处理

2.1 统一尺寸大小（yolo可不用）

yolov3有自带的统一大小的步骤。这部分可以不用。其他网络按需选择。

import cv2
import os

fullfilename=[]
filepath = "F:/PycharmProjects/image"   # 不能包含中文路径
filepath1 = "F:/PycharmProjects/resize"
for filename in os.listdir(filepath):
    print(filename)
    print(os.path.join(filepath, filename))
    filelist = os.path.join(filepath, filename)
    fullfilename.append(filelist)

i = 1
for imagename in fullfilename:
    img = cv2.imread(imagename)
    img = cv2.resize(img, (416, 416))   
    resizename = str(i)+'.jpg'          # 命名形式为1,2,3...当然啦也可以根据你喜好自己改
    isExists = os.path.exists(filepath1)
    if not isExists:
        os.makedirs(filepath1)