search

eric612 / Caffe-YOLOv3-Windows

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

The convert tools doesn't seem to work #4

Closed HKLee2040 closed 5 years ago

HKLee2040 commented 6 years ago

The convert tools doesn't seem to work. $ python create_yolo_prototxt.py yolov2.cfg yolov2.prototxt

ERROR:root:route layer is not supported ERROR:root:reorg layer is not supported ERROR:root:route layer is not supported ERROR:root:region layer is not supported

HKLee2040 commented 6 years ago

I saw there is reorg_layer in this package. Can you share the yolo prototxt example that uses reorg_layer?

eric612 commented 6 years ago

Remove region layer from .cfg , you will not see the error message. Then you can add "DetectionOutput" layer at end . you can see example prototxt at model/yolo/tiny_yolo.prototxt tiny-yolo (yolov2) download link : https://drive.google.com/open?id=1z4FkSB2e2n4Dw13Co3Uwk_Fwfcx2C0-Z

yolov2 weights and cfg without reorg layer(yolov2) download link : https://drive.google.com/open?id=1dkv0u5kKVgYJ7rqlB17r7BjCYG907MC5

If you want use original yolov2 , you need modify create_yolo_prototxt.py to identify reorg layer ,or post edit prototxt to add reorg layer self , I'm not try this way

Produce weights and models command : python create_yolo_caffemodel.py -m tiny-yolo.prototxt -w tiny-yolo-voc-custom_27000.weights -o tiny-yolo.caffemodel

requirement : need build caffe first and specify path or copy python dir. to /example

HKLee2040 commented 6 years ago

Thanks for your reply. I will try it.