search

hollance / YOLO-CoreML-MPSNNGraph

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

Boxes remain on screen after moving on from objects #24

Closed vvkv closed 6 years ago

vvkv commented 6 years ago

Thank you so much for your work here. I have began to notice the issue of bounding boxes persisting on my iphone 6s screen even after I point to another object. I am using my own ML model that I converted from custom training with darknet, I have using this model for a few days now and today is the first time this issue has come up. I reinstalled the app with my coreML graph on with no real luck yet. Any pointers as to why this might be? The app is running at about 11 FPS so its clearly not an obvious lag. Any guidance will be very helpful. Thanks

hollance commented 6 years ago

When the app gets the results from YOLO, it always reuses the old bounding boxes and then makes the remaining ones invisible. Perhaps something is wrong there? If you can reproduce the issue then that would be the first thing I'd look at.