栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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:GameHelper2yolov5-d9f23ed6d65e985c07e9ef0ec77d476dd14e2b26labelimgvideo1.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

然后就这样打标签

打完标签后就是输出了

 

 

 

 下载好后,放到这来

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/953890.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号