I am trying to train a new dataset that I’ve created. I captured images of a mask as shown in the tutorial and they are saved under jetson-inference/python/training/classification/data/dataset/train/mask. Now, I want to train the data set. So i cd to jetson-inference/python/training/classification and run the command "python3 train.py --model-dir=models/mask data/dataset/train/mask. I think I may be running the wrong command? Because terminal just returns this:
Use GPU: 0 for training
Traceback (most recent call last):
File “train.py”, line 506, in
main()
File “train.py”, line 135, in main
main_worker(args.gpu, ngpus_per_node, args)
File “train.py”, line 171, in main_worker
normalize,
File “/usr/local/lib/python3.6/dist-packages/torchvision-0.7.0a0+78ed10c-py3.6-linux-aarch64.egg/torchvision/datasets/folder.py”, line 208, in init
is_valid_file=is_valid_file)
File “/usr/local/lib/python3.6/dist-packages/torchvision-0.7.0a0+78ed10c-py3.6-linux-aarch64.egg/torchvision/datasets/folder.py”, line 94, in init
classes, class_to_idx = self._find_classes(self.root)
File “/usr/local/lib/python3.6/dist-packages/torchvision-0.7.0a0+78ed10c-py3.6-linux-aarch64.egg/torchvision/datasets/folder.py”, line 123, in _find_classes
classes = [d.name for d in os.scandir(dir) if d.is_dir()]
FileNotFoundError: [Errno 2] No such file or directory: ‘data/dataset/train/mask/train’
I think I get the same thing when I try to retrain the cat_dog dataset