Dusty-nv jetson training custom data sets generating labels

Hi @TP2049, the labels.txt file should contain one class name per line, like so:

car
bike
person

These class names should match the object classes referenced in your annotation XML files.

For the ImageSet text files, it is one image ID per line (not including the image extension). You don’t need the 1 or -1 at the end of the lines.
Ideally it would be a random split with 80% train, 10% test, 10% val (or 70/15/15 split). However for a quick test you can just duplicate them and have all the images in train/test/val.

trainval is simply the combination of train and val. However if you are doing it the quick way and duplicating the ImageSet files, don’t have two copies of all the image IDs in trainval.txt - just have it be the same as train.txt, ect.

BTW if your dataset is big you can run the pytorch-ssd code on PC/server with discrete GPU, you just need to have PyTorch and the dependencies installed there. You can export the model to ONNX on the PC - ONNX isn’t device specific. What needs done on-device is building the TensorRT engine from the ONNX model.