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

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…

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?

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.

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