revise readme

KyanChen · Feb 27, 2024 · 62223b3 · 62223b3
1 parent 866b605
commit 62223b3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/README_zh-CN.md b/README_zh-CN.md
@@ -266,7 +266,7 @@ sh ./tools/dist_test.sh configs/TTP/xxx.py ${CHECKPOINT_FILE} ${GPU_NUM}  # xxx.
 #### 单张图像预测：
 
 ```shell
-python demo/image_demo_with_cdinferencer.py ${IMAGE_FILE1} ${IMAGE_FILE2} configs/TTP/ttp_sam_large_levircd_infer.py --weights ${CHECKPOINT_FILE} --out-dir ${OUTPUT_DIR}  # IMAGE_FILE 为你想要预测的图像文件，xxx.py 为使用的配置文件，CHECKPOINT_FILE 为你想要使用的检查点文件，OUTPUT_DIR 为预测结果的输出路径
+python demo/image_demo_with_cdinferencer.py ${IMAGE_FILE1} ${IMAGE_FILE2} configs/TTP/ttp_sam_large_levircd_infer.py --checkpoint ${CHECKPOINT_FILE} --out-dir ${OUTPUT_DIR}  # IMAGE_FILE 为你想要预测的图像文件，xxx.py 为使用的配置文件，CHECKPOINT_FILE 为你想要使用的检查点文件，OUTPUT_DIR 为预测结果的输出路径
 ```
 
 

diff --git a/demo/image_demo_with_cdinferencer.py b/demo/image_demo_with_cdinferencer.py
@@ -1,3 +1,5 @@
+import sys
+sys.path.append(sys.path[0] + "/..")
 from argparse import ArgumentParser
 from opencd.apis import OpenCDInferencer
 
@@ -39,7 +41,7 @@ def main():
         args.checkpoint,
         dataset_name=args.dataset_name,
         device=args.device,
-        classes=('unchanged', 'changed'),
+        # classes=['unchanged', 'changed'],
         palette=[[0, 0, 0], [255, 255, 255]]
     )