Train_ssd.py – ModuleNotFoundError: No module named 'vision'

[leo1986]

Hi Everone!

I am follow this project Face Mask Detection System Using AI and NVIDIA Jetson Board - Hackster.io

But, when i try execute:
python3 train_ssd.py --dataset-type=voc --data=data/MaskDetector --model-dir=models/mymodel

Generate this error:
Traceback (most recent call last):
File “train_ssd.py”, line 14, in
from vision.utils.misc import str2bool, Timer, freeze_net_layers, store_labels
ModuleNotFoundError: No module named ‘vision’

I was getting the error of not finding train_ssd.py. But I managed to solve, following this…

Train_ssd.py -- Not found under jetson-inference/python/training/detection/ssd

Hi All, I am trying to train the object detection model as per the steps mentioned in the below link. Once I created the dataset and tried running python3 train_ssd.py --dataset-type=voc --data= --model-dir= but I couldn’t find any such file train_ssd.py in that directory. Can someone help me with this? Recently I cloned this repo and still am facing this issue. Is there any change in the path?

However when I found train_ssd.py it gives me this error.
~/jetson-inference/python/training/detection/ssd$ python3 train_ssd.py --dataset-type=voc --data=data/MaskDetector --model-dir=models/mymodel
Traceback (most recent call last):
File “train_ssd.py”, line 14, in
from vision.utils.misc import str2bool, Timer, freeze_net_layers, store_labels
ModuleNotFoundError: No module named ‘vision’

Someone can help-me?

[dusty_nv]

Hi @leo1986, does your jetson-inference/python/training/detection/ssd/vision directory look like the following?

https://github.com/dusty-nv/pytorch-ssd/tree/8ed842a408f8c4a8812f430cf8063e0b93a56803/vision

If not, when you cloned the repo did you use the --recursive flag? If not you can run git submodule update --init from your jetson-inference directory.