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

【问题标题】：Dataset_factory importerror: Tensorflow fine-tuning a pre-trained model from an existing checkpoint on custom dataDataset_factory importerror：Tensorflow 从自定义数据的现有检查点微调预训练模型
【发布时间】：2017-12-13 07:04:24
【问题描述】：

我正在根据 tensorflow github page 的说明在一个小型自定义数据集上重新训练一个预训练的（inception 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 行，从数据集中导入 dataset_factory - ImportError: no module named 'datasets'

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

【问题讨论】：

OP的两个链接现在都是404。2021-03-30

标签： python machine-learning tensorflow computer-vision

【解决方案1】：

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

【讨论】：

此链接现在是 404。