|
|
@@ -10,14 +10,14 @@ import onnx
|
|
|
|
|
|
from watermark_verify import verify_tool_mix
|
|
|
|
|
|
-model_types = {
|
|
|
- "classification": [
|
|
|
- "alexnet","alexnet_keras", "vgg16", "vgg16_tensorflow", "googlenet", "resnet"
|
|
|
- ],
|
|
|
- "object_detection": [
|
|
|
- "ssd", "yolox", "faster-rcnn"
|
|
|
- ],
|
|
|
-}
|
|
|
+# model_types = {
|
|
|
+# "classification": [
|
|
|
+# "alexnet","alexnet_keras", "vgg16", "vgg16_tensorflow", "googlenet", "resnet"
|
|
|
+# ],
|
|
|
+# "object_detection": [
|
|
|
+# "ssd", "yolox", "faster-rcnn"
|
|
|
+# ],
|
|
|
+# }
|
|
|
|
|
|
# 获取模型层数函数
|
|
|
def get_onnx_layer_info(onnx_path):
|
