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