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

How to save yolo-nas predictions into my workspace? #1270

Closed nahmad2000 closed 1 year ago

nahmad2000 commented 1 year ago

💡 Your Question

This is the first time I'm posting a question, so I'm sorry if I'm violating any rule

I'm using the yolo-nas code offered by Deci on a custom dataset

and i want to know how can i save the label predictions into my workspace

I don't want to save images containing detection boxes I just want to save the labels (txt files)

a label (txt file) should contain: class x-center y-center width height

Versions

No response

Louis-Dupont commented 1 year ago

Hi @ahmad2246 We don't have a pre-made script for this, but you can build one on top of SG.

I invite you to have a look at this page, it explains how to iterate over each predicted image. With this, you can easily do any filtering/transformation you want over the predictions to fit your need. Hoping this helps