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
gaoqiong
MIGraphX
Commits
46b0c33b
Unverified
Commit
46b0c33b
authored
Dec 10, 2021
by
Cagri
Committed by
GitHub
Dec 10, 2021
Browse files
Updates to examples (#1022)
nfnet update 3dunet requirements via pip 3dunet requirement and nb-clean
parent
2e337c7f
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
16 additions
and
3 deletions
+16
-3
examples/vision/python_3dunet/3dunet_inference.ipynb
examples/vision/python_3dunet/3dunet_inference.ipynb
+10
-0
examples/vision/python_nfnet/nfnet_inference.ipynb
examples/vision/python_nfnet/nfnet_inference.ipynb
+4
-2
examples/vision/python_nfnet/requirements_nfnet.txt
examples/vision/python_nfnet/requirements_nfnet.txt
+2
-1
No files found.
examples/vision/python_3dunet/3dunet_inference.ipynb
View file @
46b0c33b
...
...
@@ -10,6 +10,16 @@
"https://github.com/naomifridman/Unet_Brain_tumor_segmentation"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "09ceec31",
"metadata": {},
"outputs": [],
"source": [
"!pip install SimpleITK matplotlib scikit-image"
]
},
{
"cell_type": "code",
"execution_count": null,
...
...
examples/vision/python_nfnet/nfnet_inference.ipynb
View file @
46b0c33b
...
...
@@ -17,7 +17,9 @@
"- How to optimize NFNet ONNX model with AMD MIGraphX.\n",
"- How to run inference on AMD GPU with the optimized ONNX model.\n",
"\n",
"The NFNet utilized in this example is the smallest NFNet version, F0: 71.5M parameters (83.6% top-1 accuracy on ImageNet)"
"The NFNet utilized in this example is the smallest NFNet version, F0: 71.5M parameters (83.6% top-1 accuracy on ImageNet)\n",
"\n",
"Please make sure MIGraphX Python API is installed following the instructions at Github page."
]
},
{
...
...
@@ -107,7 +109,7 @@
"metadata": {},
"outputs": [],
"source": [
"with open('../python_
api_inference
/imagenet_simple_labels.json') as json_data:\n",
"with open('../python_
resnet50
/imagenet_simple_labels.json') as json_data:\n",
" labels = json.load(json_data)"
]
},
...
...
examples/vision/python_nfnet/requirements_nfnet.txt
View file @
46b0c33b
opencv-python
onnxruntime
image
\ No newline at end of file
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