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
## 数据集
1、真实数据
使用ImageNet数据集,并且需要转成TFRecord格式
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)
......@@ -54,15 +55,18 @@ ImageNet数据集转成TFRecord格式,可以参考以下[script](https://githu
tfrecord-imagenet
|
train-00000-of-01024
train-00000-of-01024
train-00001-of-01024
...
train-01022-of-01024
train-01023-of-01024
validation-00000-of-00128
validation-00001-of-00128
...
validation-00126-of-00128
validation-00127-of-00128
```
2、合成数据
基于随机合成的数据,不需要下载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