代码地址https://github.com/tensorflow/models/tree/master/research/slim
怎么训练我就不说了,网上很多资源。微调的方法在scripts文件夹下。
下面是一个flowers的微调参数,v3模型
#!/bin/bash
PRETRAINED_CHECKPOINT_DIR=/home/Documents/model/inception_v3
DATASET_DIR=/home/Documents/dataset/flowers
TRAIN_DIR=/home/Documents/model/flowers/train03
# Fine-tune only the new layers for 1000 steps.
CUDA_VISIBLE_DEVICES='0' python train_image_classifier.py \
--train_dir=${TRAIN_DIR} \
--dataset_name=flowers \
--dataset_split_name=train \
--dataset_dir=${DATASET_DIR} \
--model_name=inception_v3 \
--checkpoint_path=${PRETRAINED_CHECKPOINT_DIR}/inception_v3.ckpt \
--checkpoint_exclude_scopes=InceptionV3/Logits,InceptionV3/AuxLogits \
--trainable_scopes=InceptionV3/Logits,InceptionV3/AuxLogits \
--max_number_of_steps=5000 \
--batch_size=32 \
--learning_rate=0.01 \
--learning_rate_decay_type=exponential \
--save_interval_secs=60 \
--save_summaries_secs=60 \
--log_every_n_steps=100 \
--optimizer=rmsprop \
--weight_decay=0.00004
验证准确率为
开始下一步,迁移到CUB_200_2011上。