diff --git a/models/Xception/pokedex_Xception.onnx b/models/Xception/pokedex_Xception.onnx new file mode 100644 index 0000000000000000000000000000000000000000..87a4669d62753d3a3456a4be322a0abaf60886c9 Binary files /dev/null and b/models/Xception/pokedex_Xception.onnx differ diff --git a/python/convert_onnx.py b/python/convert_onnx.py index a90d6d1b941da7cd367a300c02670a3393ddd9ff..041d7fa0926e5b9ee6fd5cb5c21c15bbffc0101b 100644 --- a/python/convert_onnx.py +++ b/python/convert_onnx.py @@ -16,7 +16,7 @@ if args.model == "1": size_2=(1, 224, 224, 3) elif args.model == "2": 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_2=(1, 256, 256, 3)