164 lines
5.5 KiB
Bash
Executable File
164 lines
5.5 KiB
Bash
Executable File
#!/bin/bash
|
||
|
||
usage() {
|
||
echo "Usage: $0 l <ori_label_directory> [ -h ]"
|
||
echo "对label图片进行处理及转化（-l不能为空） "
|
||
echo "-l:原始label的路径，-h:帮助"
|
||
echo "e.g. 4_rebuild_labels.sh -l ./C组标注图片"
|
||
echo "接下来一路回车就ok"
|
||
}
|
||
|
||
ori_label_directorys=""
|
||
|
||
while getopts "hl:" opt; do
|
||
case $opt in
|
||
h)
|
||
usage
|
||
exit 0
|
||
;;
|
||
l)
|
||
ori_label_directorys=$OPTARG
|
||
;;
|
||
*)
|
||
echo -e '\033[31m!!! Error, Illegal input !!!\033[0m'
|
||
usage
|
||
exit 1
|
||
;;
|
||
esac
|
||
done
|
||
|
||
# 判断输入地址是否为空
|
||
if [ -z "$ori_label_directorys" ]; then
|
||
echo -e "\033[31m输入地址 -i -l 存在空地址\033[0m"
|
||
usage
|
||
exit 1
|
||
fi
|
||
|
||
# 地址转化
|
||
ori_label_directory=$(readlink -f "$ori_label_directorys")
|
||
if [ -z "$ori_label_directory" ]; then
|
||
echo -e "\033[31m无法解析地址，程序退出\033[0m"
|
||
echo -e "\033[31mori_label_directory\033[0m: $ori_label_directorys"
|
||
exit 1
|
||
fi
|
||
if [ ! -d "$ori_label_directory" ]; then
|
||
echo -e "\033[31mlabel目录不存在，程序退出\033[0m"
|
||
echo -e "\033[31mori_label_directory\033[0m: $ori_label_directorys"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "\033[32m_____ 4_rebuild_labels.sh _____\033[0m"
|
||
|
||
|
||
echo -n "请选择label图片搜索深度(默认为1):"
|
||
read -r fold_search_depth
|
||
if [ -z $fold_search_depth ]; then
|
||
fold_search_depth='1'
|
||
fi
|
||
|
||
save_pro_folds=""${ori_label_directory%/}"_pro_label_fold" # 去掉末尾的足/
|
||
save_pro_fold=$(readlink -f "$save_pro_folds")
|
||
echo -n "请选择初步处理label后label_pro图片存储位置(默认为$save_pro_fold):"
|
||
read -r save_pro_folds
|
||
if [ -z $save_pro_folds ]; then
|
||
save_pro_folds=""${ori_label_directory%/}"_pro_label_fold"
|
||
fi
|
||
save_pro_fold=$(readlink -f "$save_pro_folds")
|
||
|
||
echo -n "请选择label_pro图片后缀(默认为\"_label\"):"
|
||
read -r pro_suffix_name
|
||
if [ -z $pro_suffix_name ]; then
|
||
pro_suffix_name='_label'
|
||
fi
|
||
|
||
save_GT_folds=""${ori_label_directory%/}"_GT_label_fold"
|
||
save_GT_fold=$(readlink -f "$save_GT_folds")
|
||
echo -n "请选择处理label_pro后label_GT图片存储位置(默认为$save_GT_fold):"
|
||
read -r save_pro_folds
|
||
if [ -z $save_pro_folds ]; then
|
||
save_GT_folds=""${ori_label_directory%/}"_GT_label_fold"
|
||
fi
|
||
save_GT_fold=$(readlink -f "$save_GT_folds")
|
||
|
||
echo -n "请选择label_GT图片后缀(默认为\"_gtFine_labelTrainIds\"):"
|
||
read -r GT_suffix_name
|
||
if [ -z $GT_suffix_name ]; then
|
||
GT_suffix_name='_gtFine_labelTrainIds'
|
||
fi
|
||
|
||
echo -n "请选择label_GT图片通道数(1或3，默认为1):"
|
||
read -r GT_channel
|
||
if [ -z $GT_channel ]; then
|
||
GT_channel='1'
|
||
fi
|
||
if [[ $GT_channel != '1' && $GT_channel != '3' ]]; then
|
||
echo -e "\033[35mGT_channel只能为1或3，输入有误，将其默认变为1\033[0m"
|
||
GT_channel='1'
|
||
fi
|
||
|
||
echo -n "请选择GT图片背景颜色(0或255，默认为0（黑色）)):"
|
||
read -r back_gnd_color
|
||
if [ -z $back_gnd_color]; then
|
||
back_gnd_color='0'
|
||
fi
|
||
|
||
echo -n "请选择GT图片中第一类的颜色(黑色背景(0)下默认为1，白色背景(255)下默认为0):"
|
||
read -r first_class_color
|
||
if [ -z $first_class_color]; then
|
||
if [ $back_gnd_color == '255' ]; then
|
||
first_class_color='0'
|
||
else
|
||
first_class_color='1'
|
||
fi
|
||
fi
|
||
|
||
echo -n "请选择图片类型(png或jpg，默认为png（没有\".\"）)):"
|
||
read -r pic_type
|
||
if [ -z $pic_type ]; then
|
||
pic_type='png'
|
||
fi
|
||
|
||
echo -n "请选择重建起始目录(label或pro，默认为label):"
|
||
read -r Rebuild_from
|
||
if [ -z $Rebuild_from ]; then
|
||
Rebuild_from='label'
|
||
fi
|
||
if [[ $Rebuild_from != 'label' && $Rebuild_from != 'pro' ]]; then
|
||
echo -e "\033[35mRebuild_from只能为label或pro，输入有误，将其默认变为label\033[0m"
|
||
Rebuild_from='label'
|
||
fi
|
||
|
||
echo -n "请选择重建最终目标(pro或GT，默认为GT):"
|
||
read -r Rebuild_to
|
||
if [ -z $Rebuild_to ]; then
|
||
Rebuild_to='GT'
|
||
fi
|
||
if [[ $Rebuild_to != 'GT' && $Rebuild_to != 'pro' ]]; then
|
||
echo -e "\033[35mRebuild_to只能为GT或pro，输入有误，将其默认变为GT\033[0m"
|
||
Rebuild_to='GT'
|
||
fi
|
||
|
||
echo -n "请选择是否保存pro图片生成中间状态（e.g.灰度图等）(false或true，默认为false):"
|
||
read -r save_process_pics
|
||
if [ -z $save_process_pics ]; then
|
||
save_process_pics='false'
|
||
fi
|
||
if [[ $save_process_pics != 'true' && $save_process_pics != 'false' ]]; then
|
||
echo -e "\033[35msave_process_pics只能为true或false，输入有误，将其默认变为false\033[0m"
|
||
save_process_pics='false'
|
||
fi
|
||
|
||
# 获取当前脚本的路径和名称
|
||
script_path=$(dirname "$0")
|
||
# 将当前目录更改为脚本所在的路径
|
||
cd "$script_path"
|
||
|
||
# 激活conda环境
|
||
source /home/"$USER"/miniconda/bin/activate Deal_pics
|
||
echo -e "\033[35m运行：\033[0mpython 4_deal_labels.py -src_fold $ori_label_directory -save_pro_fold $save_pro_fold -save_GT_fold $save_GT_fold -fold_search_depth $fold_search_depth -pro_suffix_name $pro_suffix_name -GT_suffix_name $GT_suffix_name -GT_channel $GT_channel -back_gnd_color $back_gnd_color -first_class_color $first_class_color -pic_type $pic_type -Rebuild_from $Rebuild_from -Rebuild_to $Rebuild_to -save_process_pics $save_process_pics "
|
||
echo ""
|
||
python 4_deal_labels.py -src_fold $ori_label_directory -save_pro_fold $save_pro_fold -save_GT_fold $save_GT_fold -fold_search_depth $fold_search_depth -pro_suffix_name $pro_suffix_name -GT_suffix_name $GT_suffix_name -GT_channel $GT_channel -back_gnd_color $back_gnd_color -first_class_color $first_class_color -pic_type $pic_type -Rebuild_from $Rebuild_from -Rebuild_to $Rebuild_to -save_process_pics $save_process_pics
|
||
|
||
echo "4_rebuild_label_pics.sh重构完毕"
|
||
|