search

eric612 / Caffe-YOLOv3-Windows

A windows caffe implementation of YOLO detection network
Other
212 stars 128 forks source link

训练2类目标 #10

Closed wait1988 closed 5 years ago

wait1988 commented 5 years ago

你好，请问用yolov3_lite训练两类目标的时候，网络配置文件该如何修改呢？ 我改了conv20,conv21的num_output为18 Yolov3Loss1，Yolov3Loss2的num_class为2 训练报错： yolov3_layer.cpp:143] Check failed: input_count == tmp_input_count (6498 vs. 7581) 求正确的改法~

eric612 commented 5 years ago

@wait1988 num_output 應為 (4+1+2)*3 = 21

wait1988 commented 5 years ago

@eric612 谢谢 傻了我 手误。。。

18022443868 commented 5 years ago

目标探测训练2类，Check failed: input_count == tmp_input_count (27075 vs. 7581)，感觉num_output 有好几处，都要进行修改？