YOLACT pytorch模型转tensorflow savedModel格式
YOLACT: pytorch模型 -> onnx -> tensorflow savedModel模型。
pytorch源码需要修改部分代码才能转onnx,其中decode和nms需要用tf自己在实现。
下面代码是decode和nms以及转savedModel格式:
import time
import cv2
import tensorflow as tf
import numpy as np
MEANS = np.array([103.94, 116.78, 123.68])[N
原创
2020-08-10 15:15:48 ·
1030 阅读 ·
0 评论