Closed Flova closed 2 years ago
It would be nice to return the class index of the segmentation as an unit8 for performance reasons. This should lower USB copy times from NCS2 and improve overall performance, as well as simplify the code in the bitbots_vision.
Add an uint8 typecast to the following line. https://github.com/bit-bots/YOEO/blob/cb5f2820102271ab00d6ed914faa2c90e1d59644/yoeo/models.py#L182
It would be nice to return the class index of the segmentation as an unit8 for performance reasons. This should lower USB copy times from NCS2 and improve overall performance, as well as simplify the code in the bitbots_vision.
Proposed change
Add an uint8 typecast to the following line. https://github.com/bit-bots/YOEO/blob/cb5f2820102271ab00d6ed914faa2c90e1d59644/yoeo/models.py#L182