YOLOV5目标检测之常数设置和数据集准备

打开train后主要是把上面的default是删开，下面这个改成500次，其他的一下设置主要看自己的电脑配置，配置不好设置太高跑冒烟！！

像这样建2个文件夹和一个py文件。一个放视频，一个放图片

视频是玩游戏的视频，可以用obs录一下，图片是把视频剪成一张一张的，后面训练要用

 先用这个，

import cv2

def save_image(addr, image, num):
    address = addr + 'img_' + str(num) + '.jpg'
    print(address)
    cv2.imwrite(address, image)


video_path = r'E:\GameHelper2\yolov5-d9f23ed6d65e985c07e9ef0ec77d476dd14e2b26\labelimg\video\1.mkv'
out_path = r'E:/GameHelper2/yolov5-d9f23ed6d65e985c07e9ef0ec77d476dd14e2b26/labelimg/images/'

is_all_frame = True
sta_frame = 1     起始帧
end_frame = 1000   结束

time_interval = 8   间隔时间

videocapture = cv2.VideoCapture(video_path)

success, frame = videocapture.read()

i = 0
j = 0
while success:
    i += 1
    if 1 % time_interval == 0:
        if not is_all_frame:
            if sta_frame <= i <= end_frame:
                j += 1
                print('save frame', j)
                save_image(out_path, frame, j)
            elif i > end_frame:
                break

        else:
            j += 1
            print('save frame', j)
            save_image(out_path, frame, j)

    success, frame = videocapture.read()

 来到这网站给每张图片打标签。

然后回到文件夹新建文件夹

 

 然后改名为labels

然后回到网站点这里

 

 

然后点start project 再点 create labels list 然后在点 start project

然后就这样打标签

打完标签后就是输出了

 

 

 

 下载好后，放到这来