From c96398a227e10fa3ac0479be0994cc4f76255068 Mon Sep 17 00:00:00 2001 From: MPolaris <540492239@qq.com> Date: Sun, 27 Nov 2022 15:10:00 +0800 Subject: [PATCH] include update --- layers/__init__.py | 3 ++- utils/__init__.py | 4 +--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/layers/__init__.py b/layers/__init__.py index 60fda4e..911d400 100644 --- a/layers/__init__.py +++ b/layers/__init__.py @@ -1,4 +1,5 @@ -from utils import OPERATOR +from utils.op_registry import OPERATOR + from .conv_layers import * from .dimension_utils import * from .common_layers import * diff --git a/utils/__init__.py b/utils/__init__.py index 89af612..f65bcb1 100644 --- a/utils/__init__.py +++ b/utils/__init__.py @@ -1,6 +1,4 @@ from .onnx_loader import load_onnx_modelproto from .builder import keras_builder, tflite_builder -from .op_registry import OPERATOR -from layers import * -__all__ = ['OPERATOR', 'LOG', 'load_onnx_modelproto', 'keras_builder', 'tflite_builder'] \ No newline at end of file +__all__ = ['load_onnx_modelproto', 'keras_builder', 'tflite_builder'] \ No newline at end of file