Commit a11bd1cb authored by “qianyj”'s avatar “qianyj”
Browse files

Update README

parent 4ae16240
...@@ -46,6 +46,7 @@ pip3 install -r requirements.txt --no-deps ...@@ -46,6 +46,7 @@ pip3 install -r requirements.txt --no-deps
## 数据集 ## 数据集
1、真实数据 1、真实数据
使用ImageNet数据集,并且需要转成TFRecord格式 使用ImageNet数据集,并且需要转成TFRecord格式
ImageNet数据集可以[官网](https://image-net.org/ "ImageNet数据集官网")下载、百度搜索或者联系我们 ImageNet数据集可以[官网](https://image-net.org/ "ImageNet数据集官网")下载、百度搜索或者联系我们
ImageNet数据集转成TFRecord格式,可以参考以下[script](https://github.com/tensorflow/tpu/blob/master/tools/datasets/imagenet_to_gcs.py)[README](https://github.com/tensorflow/tpu/tree/master/tools/datasets#imagenet_to_gcspy) ImageNet数据集转成TFRecord格式,可以参考以下[script](https://github.com/tensorflow/tpu/blob/master/tools/datasets/imagenet_to_gcs.py)[README](https://github.com/tensorflow/tpu/tree/master/tools/datasets#imagenet_to_gcspy)
...@@ -54,15 +55,18 @@ ImageNet数据集转成TFRecord格式,可以参考以下[script](https://githu ...@@ -54,15 +55,18 @@ ImageNet数据集转成TFRecord格式,可以参考以下[script](https://githu
tfrecord-imagenet tfrecord-imagenet
| |
train-00000-of-01024 train-00000-of-01024
train-00000-of-01024 train-00001-of-01024
... ...
train-01022-of-01024
train-01023-of-01024 train-01023-of-01024
validation-00000-of-00128 validation-00000-of-00128
validation-00001-of-00128 validation-00001-of-00128
... ...
validation-00126-of-00128
validation-00127-of-00128 validation-00127-of-00128
``` ```
2、合成数据 2、合成数据
基于随机合成的数据,不需要下载ImageNet数据集,执行网络训练时只需要把程序执行语句中的--use_synthetic_data设置为true即可 基于随机合成的数据,不需要下载ImageNet数据集,执行网络训练时只需要把程序执行语句中的--use_synthetic_data设置为true即可
## 训练 ## 训练
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment