Img2Mol 开源项目使用教程

1. 项目目录结构及介绍

Img2Mol 项目的目录结构如下：

Img2Mol/
├── benchmark_data/
├── examples/
├── img2mol/
├── model/
├── .gitignore
├── Img2Mol.png
├── LICENSE
├── README.md
├── download_model.sh
├── environment_local-cddd.yml
├── environment.yml
├── example_inference.ipynb
├── example_inference_local_cddd.ipynb
└── setup.py

目录介绍

• benchmark_data/: 包含用于基准测试的数据。
• examples/: 包含示例图像，用于应用 Img2Mol 模型。
• img2mol/: 包含 Img2Mol 模型的 Python 模块。
• model/: 存储训练好的模型权重文件。
• .gitignore: Git 忽略文件配置。
• Img2Mol.png: 项目图标或示意图。
• LICENSE: 项目许可证文件。
• README.md: 项目说明文件。
• download_model.sh: 下载模型权重的脚本。
• environment_local-cddd.yml: 本地 CDDD 模型的环境配置文件。
• environment.yml: 项目环境配置文件。
• example_inference.ipynb: 示例推理 Jupyter Notebook。
• example_inference_local_cddd.ipynb: 使用本地 CDDD 模型的示例推理 Jupyter Notebook。
• setup.py: 项目安装脚本。

2. 项目启动文件介绍

Img2Mol 项目的主要启动文件是 example_inference.ipynb 和 example_inference_local_cddd.ipynb。这两个 Jupyter Notebook 文件展示了如何使用 Img2Mol 模型进行推理。

example_inference.ipynb

该文件展示了如何使用 Img2Mol 模型进行推理，模型权重通过网络请求加载。

example_inference_local_cddd.ipynb

该文件展示了如何使用本地 CDDD 模型进行推理，适用于无法访问网络或希望使用本地模型的场景。

3. 项目的配置文件介绍

environment.yml

该文件定义了 Img2Mol 项目所需的环境依赖。通过以下命令可以创建并激活环境：

conda env create -f environment.yml
conda activate img2mol

environment_local-cddd.yml

该文件定义了使用本地 CDDD 模型时的环境依赖。通过以下命令可以创建并激活环境：

conda env create -f environment_local-cddd.yml
conda activate img2mol

download_model.sh

该脚本用于下载 Img2Mol 模型的权重文件，并将其移动到 model/ 目录下。可以通过以下命令运行脚本：

bash download_model.sh

.gitignore

该文件定义了 Git 忽略的文件和目录，避免将不必要的文件提交到版本控制系统中。

setup.py

该文件是项目的安装脚本，可以通过以下命令安装项目：

pip install -e .

通过以上步骤，您可以顺利地配置和启动 Img2Mol 项目，并开始使用其强大的分子图像识别功能。