Detect Object Using Deep Learning. Error 'No module named torchvision.models.detection.retinanet'

[MattiaScalas]

I'm trying to use the TreeDetection deep learning model with some raster imagery (ArcGIS Pro 2.8).

I get an error 002667 Cannot load python module when I add TreeDetection.dlpk as Model Definition in the geoprocessing Detect Object Using Deep Learning. The error is:

ModuleNotFoundError: No module named 'torchvision.models.detection.retinanet'

How can I solve this problem? Is it a problem linked to my ArcGIS version?

Thanks

[DanPatterson]

were you following

Train Deep Learning Model (Image Analyst)—ArcGIS Pro | Documentation

for your version of Pro?

[PavanYadav]

@MattiaScalas Have you installed the Deep Learning Packages? if not, please download and install them (see link below) and retry. 

https://github.com/Esri/deep-learning-frameworks