assert(features[0].count == 255 * 13 * 13) crash

[moderateepheezy]

Hi @hollance, thank you for this wonderful demo.

I tried to add the implementation you did for YOLO to my own model, and the line below crashed in the YOLO.swift file.

assert(features[0].count == 255 * 13 * 13)

If I comment out the assert, I can see the bounding box in the app, but are not placed correctly.

Please what do I need to look into that could cause the assert method to fail, and also why the bounding box are not correctly placed on the object I'm detecting.

Thank you.

[hollance]

Your YOLO model probably doesn't have a 13x13 output grid, or it has fewer detectors per grid, or it doesn't have the same number of classes. You'll need to tweak these numbers in the code to suit your own version of YOLO.