Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Converting YoloV3 weights to caffe #20

Open
cpagravel opened this issue Jun 28, 2018 · 6 comments
Open

Converting YoloV3 weights to caffe #20

cpagravel opened this issue Jun 28, 2018 · 6 comments

Comments

@cpagravel
Copy link

cpagravel commented Jun 28, 2018

I'm using your script from create_yolo_caffemodel.py to convert the new YoloV3 model to a caffemodel and prepare it for the Movidius stick, but I'm getting the following error:
"ERROR: size mismatch: 61771997"

Would you know why I'm getting this error? I took the model from here as described on the author's website. Then I took the yolov3.cfg file from his repo.

@duangenquan
Copy link
Owner

duangenquan commented Jun 28, 2018

I am afraid that some layers in yolov3 are not supported, such as "route","upsample" etc. You might need to update create_yolo_caffemodel.py.

@cpagravel
Copy link
Author

I found an implementation for converting from darknet to caffe here. It doesn't have a definition for the "Yolo" type layers. I could work on porting that code to modify "create_yolo_caffemodel.py", but I'm not sure how to go about converting the Yolo layers over. Would you be able to help?

@duangenquan
Copy link
Owner

If NCSDK shows the sources codes of layers and they way to compile APIs, we can try to convert the customized layers in Yolo to NCSD. But I could not find them in their repo.

How urgent and important do you need to support the customized layers in yolov3 for NCSDK?

If you really need to support yolov3, you can try to separate one yolov3 model into several parts by the 'YOLO' layer. But this will be hard to maintain different models in future.

@cpagravel
Copy link
Author

I would really like to run yolov3 on NCS. I think a lot of people would like it since the results shown in the yolov3 paper are very promising. The Tiny-YOLO network isn't super accurate. I am most interested in seeing how many FPS can be obtained on a full sized YOLOv3 network on the NCS.

@anguoyang
Copy link

can anyone turn yolo v3 models into caffe? I could pay for it

@shartoo
Copy link

shartoo commented Oct 26, 2018

@anguoyang this repo can make darknet ran faster on ARM with cpu only,which may help.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants