topic Re: Detect Object Using Deep Learning. Error 'No module named torchvision.models.detection.retinanet' in ArcGIS Image Analyst Questions

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

MattiaScalas — Fri, 03 Feb 2023 18:05:58 GMT

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

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

DanPatterson — Fri, 03 Feb 2023 21:55:45 GMT

were you following

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

for your version of Pro?

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

PavanYadav — Wed, 22 Mar 2023 16:59:30 GMT

@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

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

ShariF — Wed, 31 Jul 2024 15:50:15 GMT

Hi.

I was having the same issue. I was working in a cloned environment under Package Manger and kept getting errors. I realized when I read the fine print of installing the deep learning frameworks, that the installation from github is updating the non-cloned environment. I had always read to NOT touch the arcgispro-py3 environment or there would be dire consequences. I do not get errors now. Good luck and I hope you have success with the Tree Detection. I'm trying out the Tree Segmentation for the first time.