Problem with Re-training SSD-Mobilenet

Hi guys, I am ultra super new in the AI world and I am struggling with just following the “Hello world” tutorial from Dusty. Does anyone knows what is going wrong in here?

Error:
martin@Jetsonnano:~/Downloads/jetson-inference/python/training/detection/ssd$ python3 train_ssd.py --dataset-type=voc --data=data/pokemons --model-dir=pokemons
2022-03-03 17:02:10 - Using CUDA…
2022-03-03 17:02:10 - Namespace(balance_data=False, base_net=None, base_net_lr=0.001, batch_size=4, checkpoint_folder=‘pokemons’, dataset_type=‘voc’, datasets=[‘data/pokemons’], debug_steps=10, extra_layers_lr=None, freeze_base_net=False, freeze_net=False, gamma=0.1, lr=0.01, mb2_width_mult=1.0, milestones=‘80,100’, momentum=0.9, net=‘mb1-ssd’, num_epochs=30, num_workers=2, pretrained_ssd=‘models/mobilenet-v1-ssd-mp-0_675.pth’, resume=None, scheduler=‘cosine’, t_max=100, use_cuda=True, validation_epochs=1, weight_decay=0.0005)
2022-03-03 17:02:10 - Prepare training datasets.
2022-03-03 17:02:10 - VOC Labels read from file: (‘BACKGROUND’, ‘Pikachu’, ‘Charmander’, ‘Putin’, ‘Biden’)
2022-03-03 17:02:10 - Stored labels into file pokemons/labels.txt.
2022-03-03 17:02:10 - Train dataset size: 16
2022-03-03 17:02:10 - Prepare Validation datasets.
2022-03-03 17:02:10 - VOC Labels read from file: (‘BACKGROUND’, ‘Pikachu’, ‘Charmander’, ‘Putin’, ‘Biden’)
2022-03-03 17:02:10 - Validation dataset size: 16
2022-03-03 17:02:10 - Build network.
2022-03-03 17:02:10 - Init from pretrained ssd models/mobilenet-v1-ssd-mp-0_675.pth
Traceback (most recent call last):
File “train_ssd.py”, line 309, in
net.init_from_pretrained_ssd(args.pretrained_ssd)
File “/home/martin/Downloads/jetson-inference/python/training/detection/ssd/vision/ssd/ssd.py”, line 119, in init_from_pretrained_ssd
state_dict = torch.load(model, map_location=lambda storage, loc: storage)
File “/home/martin/.local/lib/python3.6/site-packages/torch/serialization.py”, line 571, in load
with _open_file_like(f, ‘rb’) as opened_file:
File “/home/martin/.local/lib/python3.6/site-packages/torch/serialization.py”, line 229, in _open_file_like
return _open_file(name_or_buffer, mode)
File “/home/martin/.local/lib/python3.6/site-packages/torch/serialization.py”, line 210, in init
super(_open_file, self).init(open(name, mode))
FileNotFoundError: [Errno 2] No such file or directory: ‘models/mobilenet-v1-ssd-mp-0_675.pth’

I really appreciate any potential help I can get!!!
Thanks!!

Hi @mnabaes, can you try running this step from the tutorial again: https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-ssd.md#setup

It should download the missing mobilenet-v1-ssd-mp-0_675.pth file to your jetson-inference/python/training/detection/ssd/models directory

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.