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

Posted 2023-03-12

【中文标题】Dataset_factory importerror：Tensorflow 从自定义数据的现有检查点微调预训练模型

【英文标题】：Dataset_factory importerror: Tensorflow fine-tuning a pre-trained model from an existing checkpoint on custom data

【发布时间】：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

【参考方案1】：

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

【讨论】：

此链接现在是 404。