search

Deci-AI / super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.
https://www.supergradients.com
Apache License 2.0
4.54k stars 496 forks source link

Neg in ONNX #1555

Closed pbonazzi closed 11 months ago

pbonazzi commented 11 months ago

https://github.com/Deci-AI/super-gradients/blob/68a96bb485cb0f411dce8c42d717da9e69362dee/src/super_gradients/training/utils/bbox_utils.py#L19C1-L19C24

Negations are not supported in certain MicroControllers / Accelerators I recommend changing this line to :

    x1y1 = points - lt
pbonazzi commented 11 months ago

It also simplifies the graph see netron.app

BloodAxe commented 11 months ago

We merged the patch, thank for contributing!