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
ViT_migraphx
Commits
7abdf740
Commit
7abdf740
authored
Jun 07, 2023
by
lijian6
Browse files
Modify model.properties
Signed-off-by:
lijian
<
lijian6@sugon.com
>
parent
ee6c50a4
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
9 additions
and
9 deletions
+9
-9
Python/README.md
Python/README.md
+5
-5
Python/infer_pytroch_migraphx.py
Python/infer_pytroch_migraphx.py
+2
-2
model.properties
model.properties
+2
-2
No files found.
Python/README.md
View file @
7abdf740
...
...
@@ -53,11 +53,11 @@ Fine-tuning时可调整epoch参数来调整模型。
| Engine | Model Path| Data | Accuracy(%) |
| :------: | :------: | :------: | :------: |
| Pyt
h
orch | models/model.onnx | daisy | 98.4 |
| Pyt
h
orch | models/model.onnx | dandelion | 98.2 |
| Pyt
h
orch | models/model.onnx | roses | 90.0 |
| Pyt
h
orch | models/model.onnx | sunflowers | 97.4 |
| Pyt
h
orch | models/model.onnx | tulips | 95.4 |
| Pytorch | models/model.onnx | daisy | 98.4 |
| Pytorch | models/model.onnx | dandelion | 98.2 |
| Pytorch | models/model.onnx | roses | 90.0 |
| Pytorch | models/model.onnx | sunflowers | 97.4 |
| Pytorch | models/model.onnx | tulips | 95.4 |
| MIGraphX | models/model.onnx | daisy | 98.4 |
| MIGraphX | models/model.onnx | dandelion | 98.8 |
| MIGraphX | models/model.onnx | roses | 91.3 |
...
...
Python/infer_pytroch_migraphx.py
View file @
7abdf740
...
...
@@ -77,11 +77,11 @@ def main(intputdir):
for
i
in
range
(
len
(
predict
)):
if
(
predict
[
i
].
numpy
()
>=
0.5
):
print
(
"pyt
h
orch class: {:10} prob: {:.3}"
.
format
(
class_indict
[
str
(
i
)],
predict
[
i
].
numpy
()))
print
(
"pytorch class: {:10} prob: {:.3}"
.
format
(
class_indict
[
str
(
i
)],
predict
[
i
].
numpy
()))
if
intputdir
.
find
(
class_indict
[
str
(
i
)]):
pytorch_match_cnt
+=
1
print
(
"Pyt
h
orch Img_cnt: {:<5} match_cnt: {:<5} acc:{:.3}"
.
format
(
Img_cnt
,
pytorch_match_cnt
,
pytorch_match_cnt
/
Img_cnt
))
print
(
"Pytorch Img_cnt: {:<5} match_cnt: {:<5} acc:{:.3}"
.
format
(
Img_cnt
,
pytorch_match_cnt
,
pytorch_match_cnt
/
Img_cnt
))
print
(
"Migraphx Img_cnt: {:<5} match_cnt: {:<5} acc:{:.3}"
.
format
(
Img_cnt
,
migraphx_match_cnt
,
migraphx_match_cnt
/
Img_cnt
))
...
...
model.properties
View file @
7abdf740
...
...
@@ -3,6 +3,6 @@ modelName=Vision_Transformer
# 模型描述
modelDescription
=
ViT是一个基于transformer的视觉图像分类模型
# 应用场景(多个标签以英文逗号分割)
appScenario
=
训练,推理,train,inference,Pyt
h
orch,MIGraphX,图像分类,C++
appScenario
=
训练,推理,train,inference,Pytorch,MIGraphX,图像分类,C++
# 框架类型(多个标签以英文逗号分割)
frameType
=
MIGraphX
,Pythorch
frameType
=
MIGraphX
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