Tensorflow 关于Deeplab输出全黑(0)输出
我想问一下关于带有mobilenetv2输出全黑(0)数组的Deeplab的问题,并确认它都是零而不是0~1。我见过很多人都有同样的问题,但我还没有找到解决办法。 1.我的分割数据是1D数组(移除的颜色贴图),所以我认为这不应该是问题,根据类别,标签在0到1之间。 2.我已经设置了正确的班数、train、val、trainval数据编号。 3.我可以训练出没有任何问题的模型,并且损失值很好。 我的列车配置:Tensorflow 关于Deeplab输出全黑(0)输出,tensorflow,deeplab,Tensorflow,Deeplab,我想问一下关于带有mobilenetv2输出全黑(0)数组的Deeplab的问题,并确认它都是零而不是0~1。我见过很多人都有同样的问题,但我还没有找到解决办法。 1.我的分割数据是1D数组(移除的颜色贴图),所以我认为这不应该是问题,根据类别,标签在0到1之间。 2.我已经设置了正确的班数、train、val、trainval数据编号。 3.我可以训练出没有任何问题的模型,并且损失值很好。 我的列车配置: export PYTHONPATH=$PYTHONPATH:pwd:pwd/deepla
export PYTHONPATH=$PYTHONPATH:pwd:pwd/deeplab
WORK_DIR="deeplab"
DATASET_DIR="datasets"
# Set up the working directories.
PQR_FOLDER="eye2"
EXP_FOLDER=${WORK_DIR}/"exp/train_on_trainval_set"
INIT_FOLDER=${WORK_DIR}/"init_models"
TRAIN_LOGDIR="${WORK_DIR}/${DATASET_DIR}/${PQR_FOLDER}/${EXP_FOLDER}/train"
DATASET="${WORK_DIR}/${DATASET_DIR}/${PQR_FOLDER}/tfrecord"
mkdir -p "${WORK_DIR}/${DATASET_DIR}/${PQR_FOLDER}/exp"
mkdir -p "${TRAIN_LOGDIR}"
NUM_ITERATIONS=10000
python3 deeplab/train.py \
--logtostderr \
--train_split="train" \
--model_variant="mobilenet_v2" \
--train_crop_size=513,513 \
--train_batch_size=4 \
--training_number_of_steps="${NUM_ITERATIONS}" \
--fine_tune_batch_norm=true \
--tf_initial_checkpoint="${INIT_FOLDER}"/model.ckpt-30000\
--train_logdir="${TRAIN_LOGDIR}" \
--dataset_dir="${DATASET}" \
--initialize_last_layer=false
可视配置:
export PYTHONPATH=$PYTHONPATH:pwd:pwd/deeplab
WORK_DIR="deeplab"
DATASET_DIR="datasets"
# Set up the working directories.
PQR_FOLDER="eye2"
EXP_FOLDER=${WORK_DIR}/"exp/train_on_trainval_set"
INIT_FOLDER=${WORK_DIR}/"init_models"
TRAIN_LOGDIR="${WORK_DIR}/${DATASET_DIR}/${PQR_FOLDER}/${EXP_FOLDER}/train"
DATASET="${WORK_DIR}/${DATASET_DIR}/${PQR_FOLDER}/tfrecord"
#mkdir -p "${WORK_DIR}/${DATASET_DIR}/${PQR_FOLDER}/exp"
#mkdir -p "${TRAIN_LOGDIR}"
#--min_resize_value=1080\
#--max_resize_value=1351\
#--vis_crop_size=513,513\
NUM_ITERATIONS=20000
python3 deeplab/vis.py \
--logtostderr \
--vis_split="val" \
--model_variant="mobilenet_v2" \
--vis_crop_size=1080,1351\
--dataset_dir="deeplab/datasets/eye2/tfrecord/"\
--vis_logdir="${TRAIN_LOGDIR}/vis" \
--checkpoint_dir="deeplab/datasets/eye2/deeplab/exp/train_on_trainval_set/train"\
--max_number_of_iterations=1\
有人能帮我吗,或者请告诉我您的成功设置,就像您的配置或数据集一样。您能尝试在vis配置中设置颜色映射,看看是否有变化吗?例如:--colormap\u type=“cityscapes”\您能否尝试在vis配置中设置一个颜色映射,以查看是否有更改?例如:--colormap_type=“城市景观”\