search

hollance / YOLO-CoreML-MPSNNGraph

Tiny YOLO for iOS implemented using CoreML but also using the new MPS graph API.
MIT License
935 stars 252 forks source link

Support for yolo v3 #36

Closed chinmay5 closed 6 years ago

chinmay5 commented 6 years ago

Is there some effort in pipeline for conversion to YOLO v3?

hollance commented 6 years ago

Not from me. ;-) YOLO is generally a lot slower than SSD(Lite) (on a MobileNet backbone) and so I'm not really in a hurry to convert it to iOS.

chinmay5 commented 6 years ago

I was wondering if things need to be done for tiny-yolo part. The v3 is fast even for tiny-yolo and was wondering if conversion can be tried

chinmay5 commented 6 years ago

Can you help me a bit. I started converting tiny-yolov3 and although the Keras conversion looks good, it failed when I tried converting it to coreml image

I am not sure what exactly is the reason for the error. I would really appreciate your help

hollance commented 6 years ago

Are you sure input1 is an image input?

chinmay5 commented 6 years ago

I just followed the steps outlined. I am not sure how to check it. Can you please help

hollance commented 6 years ago

I'm afraid I don't have the time...