1

我想问一下关于使用mobilenetv2输出全黑（0）数组的Deeplab的问题，并确认它是全零而不是0~1。我看到很多人有同样的问题，但我还没有找到解决方案。1.我的分割数据是一维数组（去除了颜色图），所以我认为这应该不是问题，标签在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\

任何人都可以帮助我，或者请告诉我您的成功设置，就像您的配置或您的数据集一样。

4

0 回答 0