Update precision and test.sh

README.md

@@ -36,7 +36,7 @@ python -m pip install -e detectron2
 
 -v 路径、docker_name和imageID根据实际情况修改
 
-```
+```vv
 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py38-latest
 
 docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
@@ -244,11 +244,17 @@ bash train.sh
 ## 推理
 
 Tips:
+测试有两种展示，一种 visualization show, 测试完成后会在 --output 路径下生成测试图片结果；
+第二种是 eval show, 测试完成后会展示测试结果数据, 没有测试图片结果展示。
+
 如需执行自己的预训练模型，请修改配置:
 ${CONFIG_FILE} yaml文件配置地址（注意修改预训练模型地址）
 ${IMAGE_PATH} 待测试数据地址
+${MODEL_PATH} 待测试预训练模型地址
+
+具体配置可参考test.sh中提供的样例
 
-样例执行步骤：
+visualization show 样例执行步骤：
 
 1. 下载CTW1500的预训练模型：
 
@@ -278,16 +284,16 @@ CTW1500上的结果展示
 
 基于backbone=R50在ctw1500上的测试结果如下表所示：
 
-|Backbone|External Data|Det-P|Det-R|Det-F1|E2E-None|E2E-Full|
+|Backbone|External Data|Det-P|Det-R|Det-F1|E2E-None|E2E-Generic|
 |:------:|:------:|:------:|:------:|:------:|:------:|:------:|
-|Res-50(ours)|Synth150K+Total-Text+MLT17+IC13+IC15|0.9325|0.8475|0.8879|0.6408|0.812|
-|Res-50|Synth150K+Total-Text+MLT17+IC13+IC15|0.932|0.85|0.889|0.642|0.814|
+|Res-50(ours)|Synth150K+Total-Text+MLT17+IC13+IC15|0.9364|0.8278|0.8788|0.6839|0.9485|
+|Res-50|Synth150K+Total-Text+MLT17+IC13+IC15|0.9326|0.8240|0.8750|0.6745|0.9340|
 
 ## 应用场景
 ### 算法类别
 OCR
 
-### 热点应用行业
+### 热点应用行业v
 政府,交通,物流
 
 ## 源码仓库及问题反馈

configs/R_50/CTW1500/finetune_96voc_50maxlen.yaml

@@ -13,7 +13,7 @@ MODEL:
         POINT_COORD_WEIGHT: 0.5
         POINT_TEXT_WEIGHT: 1.0 #0.5
         BOUNDARY_WEIGHT: 0.25
-  
+
 DATASETS:
   TRAIN: ("ctw1500_train_96voc",)
   TEST: ("ctw1500_test",)

test.sh

 #!/bin/bash
 echo "Testing start ..."
-
+# visualization show
 # python demo/demo.py --config-file ${CONFIG_FILE} --input ${IMAGE_PATH}
+# eval show
+# python tools/train_net.py --config-file ${CONFIG_FILE} --eval-only MODEL.WEIGHTS ${MODEL_PATH}
 
 # examples
+# visualization show
 python demo/demo.py --config-file configs/simple/test_simple.yaml --input datasets/simple/test_images
+# eval show
+#python tools/train_net.py --config-file configs/R_50/CTW1500/finetune_96voc_50maxlen.yaml --eval-only MODEL.WEIGHTS path/pretrained_model/model.pth
\ No newline at end of file