- fix the bug in test.py

- add the gitee mirror link

README.md

@@ -8,6 +8,8 @@
 
 (ECCV 2020) Hierarchical Dynamic Filtering Network for RGB-D Salient Object Detection
 
+Gitee Mirror: <https://gitee.com/p_lart/HDFNet>
+
 > Author: Lart Pang(`lartpang@163.com`)
 >
 > This is a complete, modular and easily modified code base based on PyTorch, which is suitable for the training and testing of significant target detection task model.

test.py

@@ -9,6 +9,7 @@
 import os
 import os.path as osp
 from datetime import datetime
+from distutils.util import strtobool
 
 import numpy as np
 import torch
@@ -26,23 +27,22 @@
     epilog="Enjoy the program! :)",
     allow_abbrev=False,
 )
-my_parser.version = "1.0.0"
 my_parser.add_argument("--param_path", required=True, type=str, help="自定义参数文件路径")
 my_parser.add_argument("--model", required=True, type=str, help="选择使用的模型的名字，请把对应的模型类导入到network文件夹中的`__init__.py`文件中")
 my_parser.add_argument("--testset", required=True, type=str, help="测试集路径，该路径下至少包含两个文件夹: Image, Depth")
+# https://stackoverflow.com/a/46951029
 my_parser.add_argument(
     "--has_masks",
-    default="False",
-    choices=["True", "False"],
-    type=str,
+    default=False,
+    type=lambda x: bool(strtobool(str(x))),
     help="是否存在对应的Mask数据，即`--testset`指定的路径下是否包含存放有mask文件的Mask文件夹",
 )
-my_parser.add_argument("--save_pre", default="True", choices=["True", "False"], type=str, help="是否保存测试生成的结果")
+my_parser.add_argument("--save_pre", default=False, type=lambda x: bool(strtobool(str(x))), help="是否保存测试生成的结果")
 my_parser.add_argument("--save_path", default="", type=str, help="保存测试结果的路径")
 my_parser.add_argument(
     "--data_mode", default="RGBD", choices=["RGB", "RGBD"], type=str, help="测试的是RGB数据还是RGBD数据，注意请选择使用对应任务的模型"
 )
-my_parser.add_argument("--use_gpu", default="True", choices=["True", "False"], type=str, help="测试是否使用GPU")
+my_parser.add_argument("--use_gpu", default=True, type=lambda x: bool(strtobool(str(x))), help="测试是否使用GPU")
 my_args = my_parser.parse_args()
 
 

test.sh

@@ -1,10 +1,10 @@
 #!/usr/bin/env bash
 
 python test.py --param_path /home/lart/Coding/HDFFile/output/HDFNet/PretrainedParams/HDFNet_VGG16/HDFNet_VGG16_7Datasets.pth \
-               --model HDFNet \
+               --model HDFNet_VGG16 \
                --testset /home/lart/Datasets/Saliency/RGBDSOD/LFSD/ \
                --has_masks True \
-               --save_pre False \
+               --save_pre True \
                --save_path output/HDFNet/pre/test \
                --data_mode RGBD \
                --use_gpu True