search

naseemap47 / YOLO-NAS

Train and Inference your custom YOLO-NAS model by Single Command Line
Apache License 2.0
97 stars 13 forks source link

NMS missing in inference #52

Open ghost opened 9 months ago

ghost commented 9 months ago

The training/testing allows it, but the inference script has no support for NMS, which makes me think it is not carried out. It would be nice to add it, in order to be able to test the models more efficiently.

naseemap47 commented 9 months ago

Hi @josemafuen, The training setup I used from super-gradients We have to give suggestion for them to add this feature. This is not in my hand.