Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_onnxruntime
Commits
d27b724b
Commit
d27b724b
authored
Nov 07, 2023
by
yangql
Browse files
Update README.md
parent
f279ce5f
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
18 additions
and
1 deletion
+18
-1
README.md
README.md
+18
-1
No files found.
README.md
View file @
d27b724b
...
...
@@ -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版本推理
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment