61 lines
1.6 KiB
Bash
61 lines
1.6 KiB
Bash
#!/bin/bash
|
||
# 原图像位置
|
||
Dir_src_pics="./data/img"
|
||
Dir_result="./data/result"
|
||
Dir_ori_src_pics="/root/Dehaze/SRC_files/src"
|
||
|
||
mkdir -p $Dir_src_pics $Dir_result
|
||
|
||
# 进行绝对路径转化
|
||
Dir_src_pics=$(readlink -f "$Dir_src_pics")
|
||
Dir_result=$(readlink -f "$Dir_result")
|
||
if [ ! -d "$Dir_src_pics" ] && [ ! -d "$Dir_result" ]; then
|
||
echo "image、label都不存在，程序退出"
|
||
echo -e "\033[31mori_image_directory\033[0m: $Dir_src_pics"
|
||
echo "$Dir_src_pics"
|
||
echo -e "\033[31mori_label_directory\033[0m: $Dir_result"
|
||
echo "$Dir_result"
|
||
exit 1
|
||
fi
|
||
|
||
PS3='All in one choice : '
|
||
applications=("Delete_src_pics" "Delete_generate_pics" "Copy_src_pics_to_test" "Run_test_program" "quit")
|
||
select fav in "${applications[@]}"; do
|
||
case $fav in
|
||
# 删除原始文件选项
|
||
"Delete_src_pics")
|
||
# 删除src文件
|
||
echo "Delete all src files in $Dir_src_pics"
|
||
rm $Dir_src_pics/*
|
||
;;
|
||
|
||
# 删除生成文件选项
|
||
"Delete_generate_pics")
|
||
# 删除result文件
|
||
echo "Delete all src files in $Dir_result"
|
||
rm $Dir_result/*
|
||
;;
|
||
|
||
# 复制待处理文件选项
|
||
"Copy_src_pics_to_test")
|
||
# 删除src文件
|
||
echo "Copy all src files in $Dir_ori_src_pics"
|
||
ln -s $Dir_ori_src_pics/* $Dir_src_pics
|
||
;;
|
||
|
||
# 运行程序
|
||
"Run_test_program")
|
||
source ~/miniconda/bin/activate Dehaze_DCP
|
||
python ./test/test.py $Dir_src_pics $Dir_result
|
||
;;
|
||
|
||
# 退出选项
|
||
"quit")
|
||
echo "User requested exit"
|
||
exit
|
||
;;
|
||
|
||
# 其他选项
|
||
*) echo "invalid option $REPLY";;
|
||
esac
|
||
done |