Python Dataset_factory importerror:Tensorflow从自定义数据的现有检查点微调预先训练的模型
我正在根据tensorflow github的说明,在一个小型定制数据集上重新训练一个预先训练过的(inception v1)模型 创建数据集:Python Dataset_factory importerror:Tensorflow从自定义数据的现有检查点微调预先训练的模型,python,machine-learning,tensorflow,computer-vision,Python,Machine Learning,Tensorflow,Computer Vision,我正在根据tensorflow github的说明,在一个小型定制数据集上重新训练一个预先训练过的(inception v1)模型 创建数据集: python build_image_data.py --train_directory="${TRAIN_DIR}" --validation_directory="${VALIDATION_DIR}" --output_directory="${OUTPUT_DIRECTORY}" --labels_file="${LABELS_FILE}
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
微调(-Fine tuning a model from a existing checkpoint section):
我收到了以下错误消息:
文件“train_image_classifier.py”,第23行,来自数据集导入数据集工厂-导入错误:没有名为“数据集”的模块
我试图将dataset_工厂添加到python路径中,但它不起作用,我找不到任何解决方案来解决这个问题。在这种情况下该怎么办?首先,让舒尔知道您拥有最新版本的代码。检查数据集文件夹是否存在。您还可以尝试将dataset(而不是dataset_factory.py)文件夹添加到PYTHONPATH环境变量。OP的两个链接现在都是404。2021-03-30此链接现在是404。
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}"