50 lines
1.4 KiB
Bash
50 lines
1.4 KiB
Bash
#!/bin/bash
|
|
# 原图像位置
|
|
Dir_pic_root="./datasets/quick_test"
|
|
Dir_src_pics="./datasets/quick_test/src"
|
|
Dir_result="./datasets/quick_test/result"
|
|
Dir_ori_src_pics="/root/Dehaze/SRC_files/src"
|
|
|
|
mkdir -p $Dir_pic_root $Dir_src_pics $Dir_result
|
|
|
|
PS3='All in one choice : '
|
|
applications=("Delete_src_pics" "Delete_generate_pics" "Copy_src_pics" "Run_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")
|
|
# 删除src文件
|
|
echo "Copy all src files in $Dir_ori_src_pics"
|
|
ln -s $Dir_ori_src_pics/* $Dir_src_pics
|
|
;;
|
|
|
|
# 运行程序
|
|
"Run_program")
|
|
source ~/miniconda/bin/activate Dehaze_GCANet
|
|
python quick_test.py --dataroot $Dir_pic_root --dataset_mode single --name refined_DCP_outdoor --model refined_DCP --phase src --preprocess none --save_image --method_name refined_DCP_outdoor_ep_60 --epoch 60
|
|
;;
|
|
|
|
# 退出选项
|
|
"quit")
|
|
echo "User requested exit"
|
|
exit
|
|
;;
|
|
|
|
# 其他选项
|
|
*) echo "invalid option $REPLY";;
|
|
esac
|
|
done |