add pt2tf tool

This commit is contained in:
zxros10
2020-09-23 09:09:49 +08:00
parent 7f7b7df65d
commit 18aefa4dd0
407 changed files with 16211 additions and 0 deletions
@@ -0,0 +1,19 @@
import tensorflow as tf
from onnx_tf.handlers.backend_handler import BackendHandler
from onnx_tf.handlers.handler import onnx_op
@onnx_op("ImageScaler")
class ImageScaler(BackendHandler):
@classmethod
def version_1(cls, node, **kwargs):
input_dict = kwargs["tensor_dict"]
x = input_dict[node.inputs[0]]
scale = node.attrs.get("scale", 1.0)
output = tf.multiply(x, scale)
if "bias" in node.attrs:
bias = node.attrs["bias"]
output = tf.nn.bias_add(output, bias, data_format="NCHW")
return [output]