I am training the resnet-18 model on a 4 new classes. I followed the steps here https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-collect.md. The issue is that after completing all steps and attempting to run imagenet-console with this command “imagenet-console --model=RoadDetection/resnet18.onnx --input_blob=input_0 --output_blob=output_0 --labels=$DATASET/labels.txt $DATASET/P1130932.JPG tested.JPG” I get the following errors:
imageNet – loaded 4 class info entries
imageNet – didn’t load expected number of class descriptions (4 of 1)
imageNet – failed to load synset class descriptions (4 / 4 of 1)
Searching online does not help as nobody else seems to have had this particular issue. Any help that can be provided is appreciated.