Select to view content in your preferred language

Integrating Torch and Pickle model files with GeoAI Tools

108
1
2 weeks ago
Status: Already Offered
Labels (1)
MuhammadElbagoury
Occasional Contributor

Hello,
I think it would be great benefit if ArcGIS Pro Deep Learning Tools become able to read model files like .pt or pickle files without the need to hardcode this to convert these formats to .emd first.
Thanks.

1 Comment
PriyankaTuteja
Status changed to: Already Offered

Hello @MuhammadElbagoury ,
Thank you for your suggestion!

ArcGIS tools already support .pt (PyTorch) and other pickled model files. To use externally trained models within ArcGIS, you'll need to package them into a Deep Learning Package (.dlpk). This is a structured archive containing the model definition, weights, and necessary configuration.

The process typically involves:

  1. Creating an .emd (Esri Model Definition) file – This file describes the model's inputs, outputs, and configuration parameters.
  2. Organizing the required files – Arrange the .emd file, model weights (e.g., .pt), and any supporting files in the correct directory structure.
  3. Packaging everything into a .dlpk – Compress the structured directory into a .dlpk file for use in ArcGIS.

For detailed guidance, including examples and templates, you can refer to the Esri Custom Models GitHub repository.