Skip to content

Commit 46b0c33

Browse files
author
Cagri
authored
Updates to examples (#1022)
nfnet update 3dunet requirements via pip 3dunet requirement and nb-clean
1 parent 2e337c7 commit 46b0c33

File tree

3 files changed

+16
-3
lines changed

3 files changed

+16
-3
lines changed

examples/vision/python_3dunet/3dunet_inference.ipynb

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,16 @@
1010
"https://github.com/naomifridman/Unet_Brain_tumor_segmentation"
1111
]
1212
},
13+
{
14+
"cell_type": "code",
15+
"execution_count": null,
16+
"id": "09ceec31",
17+
"metadata": {},
18+
"outputs": [],
19+
"source": [
20+
"!pip install SimpleITK matplotlib scikit-image"
21+
]
22+
},
1323
{
1424
"cell_type": "code",
1525
"execution_count": null,

examples/vision/python_nfnet/nfnet_inference.ipynb

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,9 @@
1717
"- How to optimize NFNet ONNX model with AMD MIGraphX.\n",
1818
"- How to run inference on AMD GPU with the optimized ONNX model.\n",
1919
"\n",
20-
"The NFNet utilized in this example is the smallest NFNet version, F0: 71.5M parameters (83.6% top-1 accuracy on ImageNet)"
20+
"The NFNet utilized in this example is the smallest NFNet version, F0: 71.5M parameters (83.6% top-1 accuracy on ImageNet)\n",
21+
"\n",
22+
"Please make sure MIGraphX Python API is installed following the instructions at Github page."
2123
]
2224
},
2325
{
@@ -107,7 +109,7 @@
107109
"metadata": {},
108110
"outputs": [],
109111
"source": [
110-
"with open('../python_api_inference/imagenet_simple_labels.json') as json_data:\n",
112+
"with open('../python_resnet50/imagenet_simple_labels.json') as json_data:\n",
111113
" labels = json.load(json_data)"
112114
]
113115
},
Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
opencv-python
2-
onnxruntime
2+
onnxruntime
3+
image

0 commit comments

Comments
 (0)