摘要: .pt和.pth文件都是用于保存PyTorch模型的文件格式,习惯上: .pt是完整的模型(full model):包括模型的架构和权重。 .pth只有模型的权重(weights) .pt的保存与加载 import torch # 假设有一个训练好的模型 model torch.save(model 阅读全文
posted @ 2022-08-11 15:47 夕西行 阅读(344) 评论(0) 推荐(0) 编辑
摘要: OpenCV里可以直接使用的模型: readNetFromTorch # 注意是torch而不是pytorch,所以要的模型是.t7格式的。 readNetFromTensorflow readNetFromONNX readNetFromCaffe readNetFromDarknet 案例: O 阅读全文
posted @ 2022-08-11 15:38 夕西行 阅读(1139) 评论(0) 推荐(0) 编辑