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

Error when converting YOLOv2 model to Metal format. #38

Closed tkreiman closed 6 years ago

tkreiman commented 6 years ago

When I try running the nngraph.py script, I get an error in line 28 saying that the index is out of range. Is there no batch normalization layer here? How do I fix this?

tkreiman commented 6 years ago

Just had to change the specific layers from the tiny model to the corresponding yolo model.