Closed JNaranjo-Alcazar closed 3 years ago
The error is related to transfering the pytorch model to keras (the toolbox uses keras internally). This model translation is handled by the external onnx / onnx2keras tool, so I won't be able to help with this. Possible workarounds: Try to find out what the offending operation ("ATen"?) does in your model and whether it can be replaced by something that onnx can handle. Or use a Keras implementation of Yolo directly.
Unrelated to this issue, but related to using Yolo: The SNN toolbox currently supports classification tasks; object detection won't work out of the box. Several users have made the necessary modifications, but I am not aware of a public branch. See related #27 #56 for pointers on what is needed.