How to import torchvision.models.detection

Autonomous Machines Jetson & Embedded Systems Jetson AGX Xavier
#1

When i want to use and import torchvision.models.detection its says that ModuleNotFoundError: No module named ‘torchvision.models.detection’

it is hard to install some libraries’ suitable version on xavier.My torchvision version is 0.2.2.post3

thanks in advance

#2

Hi @Valerien, it appears that torchvision.models.detection was not supported yet in torchvision v0.2.2:

https://github.com/pytorch/vision/tree/v0.2.2/torchvision/models

I recommend to install newer torchvision version (and if necessary, upgrade PyTorch) to get the torchvision detection models. You can find the instructions for installing PyTorch/torchvision on Jetson in this topic: