Commit d27b724b authored by yangql's avatar yangql
Browse files

Update README.md

parent f279ce5f
...@@ -37,7 +37,24 @@ docker run --shm-size 16g --network=host --name=resnet50_ort --privileged --devi ...@@ -37,7 +37,24 @@ docker run --shm-size 16g --network=host --name=resnet50_ort --privileged --devi
``` ```
## 数据集 ## 数据集
该推理模型使用Imagenet-1k数据集训练而成。 下载ImageNet数据集:[ImageNet (image-net.org)](https://image-net.org/)
参考[scrips/extract_ILSVRC.sh](https://developer.hpccube.com/codes/modelzoo/resnet50-pytorch/-/blob/main/scrips/extract_ILSVRC.sh)处理数据集,处理后数据集结构如下:
```
data
|
train
|
n01440764
n01443537
...
val
|
n01440764
n01443537
...
```
## 推理 ## 推理
### Python版本推理 ### Python版本推理
......
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