Dataset_factory importerror:Tensorflow从自定义d上的现有检查点微调预训练模型

Question

我正在研究基于TensorFlowGithubpage的指令，在一个小的自定义数据集上重新培训一个预先训练的（初始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-从现有检查点部分微调模型）：

^{pr2}$

我收到了以下错误消息：

File "train_image_classifier.py", line 23, in from datasets import dataset_factory - ImportError: no module named 'datasets'

我试图将dataset_uFactory添加到python路径，但它没有工作，因此找不到解决此问题的任何解决方案。在这种情况下该怎么办？

Answer 1

首先，让舒尔确认您拥有最新版本的代码https://github.com/tensorflow/models/tree/master/slim。检查datasets文件夹是否存在。您也可以尝试添加数据集（而不是数据集_工厂.py)文件夹到PYTHONPATH环境变量。在