pytorch快速上手-使用自动标注软件Openlabeling和yolov5快速完成目标检测

安装

自行github下载:
openlabeling
yolov5

自动标注软件openlabeling

实际上就是标注软件里面,给你内嵌一个追踪算法,可以是光流的,也可以是边缘检测的
安装的时候记得在里面放个DaSiamRPN(其他也行,自己看readme)
DaSiamRPN

然后你标注前几张 按P 就能自动标注了,甚至标注完格式什么的都是不用改的直接就用

yolov5

pytorch版的yolov5集成度很高了
基本上就加一个配置yaml就行了
pytorch快速上手-使用自动标注软件Openlabeling和yolov5快速完成目标检测_第1张图片
data里新建一个你的配置文件,改下path标注数据地址,nc分类数量,还有names分类名就行了
然后跑训练就行了

train.py --img 1920 --batch 4 --epochs 10000 --data tarkov1.yaml --weights yolov5s.pt --workers 1

这几个参数 img为最大变宽,向下兼容
batch根据gpu来定,显存炸了就改小点
权重一般用最小的s,你有钱可以用yolov5m或者l,x

你可能感兴趣的:(pytorch,深度学习,python,pytorch,深度学习,神经网络)