mmclassificatio

OpenMMLab OpenMMLab mmdeploy v1.1.0模型部署（一）

目前，我们已经支持5个算法库和5种后端推理引擎，囊括多种应用场景:MMDeploy所支持算法库：•检测（MMDetection）•分割（MMSegmentation）•分类（MMClassificatio
qq_41627642·2023-10-17 21:00

MMClassificatio 框架下 Pytorch模型转TensorRT

模型的加载importtorchvision.modelsasmodelsresnet34=models.resnet34()resnet34.load_state_dict(torch.load('latest.pth')['model'])要解决的疑问load_state_dicttorch.load作用网络结构有了这部分是在加载参数dummyinput作用给网络一个输入如果dynamic_a
奥特慢走·2023-04-06 21:32
上一页 1 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他