49 lines
1.2 KiB
Bash
49 lines
1.2 KiB
Bash
#!/bin/bash
|
|
# 原图像位置
|
|
Dir_src_pics="./imgs/src"
|
|
Dir_result="./imgs/result"
|
|
Dir_ori_src_pics="/root/Dehaze/SRC_files/src"
|
|
|
|
mkdir -p $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 test.py --task dehaze --gpu_id 0 --indir ./imgs/src --outdir ./imgs/result
|
|
;;
|
|
|
|
# 退出选项
|
|
"quit")
|
|
echo "User requested exit"
|
|
exit
|
|
;;
|
|
|
|
# 其他选项
|
|
*) echo "invalid option $REPLY";;
|
|
esac
|
|
done |