Dataset_factory importerror：Tensorflow从自定义数据的现有检查点微调预先训练的模型

Question

我正在根据tensorflow github page

的指示，在一个小型自定义数据集上重新训练预训练（初始v1）模型

创建数据集：

python build_image_data.py 
--train_directory="${TRAIN_DIR}" 
--validation_directory="${VALIDATION_DIR}" 
--output_directory="${OUTPUT_DIRECTORY}" 
--labels_file="${LABELS_FILE}" 
--train_shards=128 
--validation_shards=24 
--num_threads=8

微调（https://github.com/tensorflow/models/tree/master/slim#Pretrained - 从现有检查点部分微调模型）：

python train_image_classifier.py 
--train_dir="${TRAIN_DIR}"  
--dataset_dir="${DATASET_DIR}" 
--dataset_name=objects 
--dataset_split_name=train 
--model_name=inception_v1 
--checkpoint_path="${CHECKPOINT_PATH}"

我收到以下错误消息：

  

文件“train_image_classifier.py”，第23行，来自数据集import dataset_factory - ImportError：没有名为'datasets'的模块

我尝试将dataset_factory添加到python路径但它没有用，我找不到任何解决方案来解决这个问题。在这种情况下该怎么办？

Answer 1

首先，请确保您拥有最新版本的代码https://github.com/tensorflow/models/tree/master/slim。检查，该数据集文件夹是否存在。您也可以尝试将数据集（而不是dataset_factory.py）文件夹添加到PYTHONPATH环境变量中。