刘晨阳、陈克彦、张浩田、齐鹏飞、邹正霞、史振伟*✉
论文的官方 PyTorch 实现:《 Change-Agent: Toward Interactive Innovative Remote Sensing Change Interpretation and Analysis 》 in [IEEE] (Accepted by IEEE TGRS 2024)
MCI模型概述:
环境安装:
步骤1 :创建一个名为Multi_change_env
的虚拟环境并激活它。
conda create - n Multi_change_env python = 3.9
conda activate Multi_change_env
第 2 步:下载或克隆存储库。
git clone https : // github . com / Chen - Yang - Liu / Change - Agent . git
cd . / Change - Agent / Multi_change
步骤 3 :安装依赖项。
pip install - r requirements . txt
下载数据集:
链接:LEVIR-MCI。 LEVIR-MCI的数据结构组织如下:
├─/DATA_PATH_ROOT/Levir-MCI-dataset/
├─LevirCCcaptions.json
├─images
├─train
│ ├─A
│ ├─B
│ ├─label
├─val
│ ├─A
│ ├─B
│ ├─label
├─test
│ ├─A
│ ├─B
│ ├─label
其中文件夹A
包含前期图像,文件夹B
包含后期图像,文件夹label
包含更改检测蒙版。
提取 LEVIR-MCI 中每个图像对的描述的文本文件:
python preprocess_data.py
之后,您可以在./data/LEVIR_MCI/
中找到一些生成的文件。
确保您执行了上述数据准备。然后,开始训练如下:
python train . py - - train_goal 2 - - data_folder / DATA_PATH_ROOT / Levir - MCI - dataset / images - - savepath . / models_ckpt /
python test . py - - data_folder / DATA_PATH_ROOT / Levir - MCI - dataset / images - - checkpoint { checkpoint_PATH }
我们建议训练模型 5 次以获得平均分数。
运行推理开始如下:
python predict . py - - imgA_path { imgA_path } - - imgB_path { imgA_path } - - mask_save_path . / CDmask . png
您可以在predict.py
中修改Change_Perception.define_args()
的--checkpoint
。然后你可以使用你自己的模型,当然,你也可以在这里下载我们的预训练模型MCI_model.pth
:[拥抱脸]。之后,将其放入./models_ckpt/
中。
代理安装:
cd . / Change - Agent / lagent - main
pip install - e .[ all ]
运行代理:
cd 进入Multi_change
文件夹:
cd . / Change - Agent / Multi_change
(1) 运行Agent CLI演示:
# You need to install streamlit first
# pip install streamlit
python try_chat.py
(2) 运行代理网页演示:
# You need to install streamlit first
# pip install streamlit
streamlit run react_web_demo.py
如果您发现本文对您的研究有用,请考虑引用:
@ARTICLE{Liu_Change_Agent,
author={Liu, Chenyang and Chen, Keyan and Zhang, Haotian and Qi, Zipeng and Zou, Zhengxia and Shi, Zhenwei},
journal={IEEE Transactions on Geoscience and Remote Sensing},
title={Change-Agent: Toward Interactive Comprehensive Remote Sensing Change Interpretation and Analysis},
year={2024},
volume={},
number={},
pages={1-1},
keywords={Remote sensing;Feature extraction;Semantics;Transformers;Roads;Earth;Task analysis;Interactive Change-Agent;change captioning;change detection;multi-task learning;large language model},
doi={10.1109/TGRS.2024.3425815}}
感谢以下存储库:
RSICC前任;更改2Cap;潜伏的
该存储库是根据 MIT 许可证分发的。该代码仅可用于学术目的。
如果您还有其他疑问❓请及时联系我们?