Esri/deep-learning-frameworks

解决的问题

该项目为在ArcGIS生态系统（ArcGIS Pro、Server和ArcGIS API for Python）中运行AI工具所需的深度学习和机器学习库提供了全面的安装程序。它通过捆绑254个包（包括PyTorch、transformers、Fast.ai和scikit-learn）消除了手动配置复杂环境的工作，确保了AI框架与地理空间软件之间的兼容性。

工作原理

该安装程序将精选的深度学习包添加到默认的arcgispro-py3 Python环境中。它支持多个平台，包括Windows（通过.msi安装程序）和Linux（通过.sh脚本）。对于离线环境中的用户，它提供了单独的“骨干”包，以便在没有互联网的情况下进行模型训练。这些工具旨在通过CUDA利用NVIDIA GPU加速计算，但某些组件可以回退到CPU。

适用对象

使用ArcGIS的地理空间分析师和开发人员，需要对地理数据执行特征提取、像素分类和特征分类等AI任务。

亮点

• 广泛的库捆绑：包括254个包，涵盖PyTorch和scikit-learn等行业标准。
• 地理空间集成：直接启用arcgis.learn模块以及用于对象检测和训练的集成地理处理工具。
• 跨平台支持：为ArcGIS Pro（Windows）、ArcGIS Server（Windows/Linux）、Drone2Map和Reality Studio提供专用安装程序。
• 离线能力：为气隙或离线环境提供特定的骨干包。