Tensorflow 关于Deeplab输出全黑(0)输出

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

我想问一下关于带有mobilenetv2输出全黑(0)数组的Deeplab的问题,并确认它都是零而不是0~1。我见过很多人都有同样的问题,但我还没有找到解决办法。 1.我的分割数据是1D数组(移除的颜色贴图),所以我认为这不应该是问题,根据类别,标签在0到1之间。 2.我已经设置了正确的班数、train、val、trainval数据编号。 3.我可以训练出没有任何问题的模型,并且损失值很好。 我的列车配置:

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=“城市景观”\