![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
torch
颐水风华
框架:tensorflow pytorch tensorrt cuda cublas opnecv。
语言:c、c++ python。
展开
-
yolov5验证模型精度
yolov5验证模型代码修改原创 2022-06-19 17:10:21 · 2173 阅读 · 0 评论 -
pth2wts
import sysimport argparseimport osimport structimport torch,loggingimport torchvision.models as models# from utils.torch_utils import select_devicelogger = logging.getLogger(__name__)def select_device(device='', batch_size=None): # device = 'c.原创 2021-12-20 15:08:46 · 485 阅读 · 0 评论