如何重新训练自定义yolo重量?

时间:2020-07-01 22:23:08

标签: object-detection yolo darknet

我在YOLOv3中进行了3种以上的自定义检测,但是检测不准确,因此我想用更多图像来训练我的自定义YOLO权重,但是 当我用新图像运行它时:

!./darknet detector train data/obj.data cfg/yolov3_custom.cfg yolov3_custom_last.weights 

它立即完成了吗?我做错了什么?

obj.data的内容

classes = 3
train = data/train.txt
valid = data/test.txt
names = data/obj.names
backup = /mydrive/yolov3/backup/

yolov3_custom.cfg的内容:

# Training
batch=64
subdivisions=16
width=416
height=416
channels=3
momentum=0.9
decay=0.0005
angle=0
saturation = 1.5
exposure = 1.5
hue=.1
learning_rate=0.001
burn_in=1000
max_batches = 6000
policy=steps
steps=4800,5400
scales=.1,.1

1 个答案:

答案 0 :(得分:0)

@Hamid Shatu。你是对的。您需要更新max_batches =并检查体重文件名称。