Skip to content
Snippets Groups Projects
Commit 9170db79 authored by michael.divia's avatar michael.divia
Browse files

2 onnx

parent 3c52774f
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
models/Xception/pokedex_Xception.onnx 0 → 100644
+ 0
− 0
View file @ 9170db79
File added
python/convert_onnx.py
+ 1
− 1
View file @ 9170db79
...@@ -16,7 +16,7 @@ if args.model == "1": ...@@ -16,7 +16,7 @@ if args.model == "1":
size_2=(1, 224, 224, 3) size_2=(1, 224, 224, 3)
elif args.model == "2": elif args.model == "2":
h5_path = "../models/Xception/pokedex_Xception.h5" h5_path = "../models/Xception/pokedex_Xception.h5"
onnx_path = "../models/ResNet50/pokedex_Xception.onnx" onnx_path = "../models/Xception/pokedex_Xception.onnx"
size=(256, 256, 3) size=(256, 256, 3)
size_2=(1, 256, 256, 3) size_2=(1, 256, 256, 3)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment