update.md · 丁浩然/deep-template

1. 将`data_prepare/data_convert.py`文件中第`14`行中的`--dataset-name`改为数据集名称，如`signs`；
2. 将`data_prepare/src/tfrecord.py`文件中第`339`行中的`range(len(filenames))`改为`list(range(len(filenames)))`，即加上`list()`；
3. 将`data_prepare/src/tfrecord.py`文件中所有`xrange`（共`3`个）全部替换为`range`；
4. 将`data_prepare/src/tfrecord.py`文件中第`160`行中的`'r'`改为`'rb'`；
5. 将`data_prepare/src/tfrecord.py`文件中第`101`，`104`，`105`，`106`行中的`_bytes_feature(name),`改为`_bytes_feature(name.encode('utf-8'))`，即在`colorspace`，`text`，`image_format`，`os.path.basename(filename)`的后面加上`.encode('utf-8')`；
6. 在`slim/datasets`文件夹中，先根据`flowers.py`进行适当的修改（共`4`处），再将数据集添加至`dataset_factory.py`文件中；
7. 参照`slim/signs`目录进行相应的目录结构调整；
8. 在`slim/train_image_classifier.py`文件中第`575`行前加入`gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.4)`以及`config = tf.ConfigProto(allow_soft_placement=True, gpu_options=gpu_options)`两行代码，并在`slim.learning.train()`的最后加上`session_config=config`；
9. 在`slim/eval_image_classifier.py`文件中第`187`行前加入`gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.4)`以及`config = tf.ConfigProto(allow_soft_placement=True, gpu_options=gpu_options)`两行代码，并在`slim.evaluation.evaluate_once()`的最后加上`session_config=config`；
10. 根据参考`train.bat`、`eval.bat`、`export.bat`、`freeze.bat`、`classify.bat`中的命令一步一步地进行后续操作即可得到最终的结果。

丁浩然/deep-template

简介

发行版

贡献者

近期动态

丁浩然/deep-template .gitee-modal { width: 500px !important; }

简介

发行版

贡献者

近期动态

搜索帮助

丁浩然/deep-template