我正在使用杰森推论,并试图收集自己的数据集并训练模型。我在jetson-inference / python / training / classification / data中创建了一个名为“ example”的文件夹,并创建了txt文件。然后,我在终端上运行了camera-capture dev / video0命令。当我将路径放置到文件夹和txt文件时,出现错误消息“无法创建数据集子目录// train”“无法创建数据集子目录// val”“无法创建数据集子目录// test ”。我认为这可能是问题所在?我插入错误的路径吗?
无论如何,收集图像后,我运行命令“ python3 train.py --model-dir = models / example data / example”。但是然后我得到了:
'''
Use GPU: 0 for training
=> dataset classes: 2 ['airpods', 'lysol']
Traceback (most recent call last):
File "train.py", line 506, in <module>
main()
File "train.py", line 135, in main
main_worker(args.gpu, ngpus_per_node, args)
File "train.py", line 191, 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 100, in __init__
raise RuntimeError(msg)
RuntimeError: Found 0 files in subfolders of: data/example/val
Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,.webp
'''