How to create an AI model with annotation and image data?

I got the annotation, YOLO format, and image data.
I am supposed to use TAO or TensorRT.
The data are like these.

annotation

1 0.510417 0.605208 0.494444 0.032292
2 0.430208 0.648177 0.204861 0.051562
0 0.376042 0.559375 0.189583 0.029167

class

barcode
name
price

Is the process to create an model with annotation and image data following these?

1. Convert YOLO format to COCO or KITTI format
2. Write spec.yaml
3. tao train

   $ ngc registry model download-version 
   nvidia/tao/pretrained_classification_tf2:efficientnet_b0
   $ tao classification_tf2 train -e /path/to/spec.yaml 
   

4. tao evaluate

   $ tao classification_tf2 evaluate -e /path/to/spec.yaml
   

And I do not know how I should write spec.yaml with the annotation data.

Any advice is appreciated.

Ubuntu: 22.04
TAO Toolkit: 5.3

2 posts - 2 participants

Read full topic