- tfrecord的生成代码
python deeplab/datasets/build_voc2012_data.py \
--image_folder="/home/dsx/ccc/img" \
--semantic_segmentation_folder="/home/dsx/ccc/mask" \
--list_folder="/home/dsx/ccc/index" \
--image_format="png" \
--output_dir="/home/dsx/ccc/tfrecord"
- folder含义
--image_folder=图片路径
--semantic_segmentation_folder=图片掩码(mask)路径
--list_folder=图片list路径
--image_format=图片格式
--output_dir=tfrecord路径
- train.py 修改
initialize_last_layer=False
la