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

浙公网安备 33010602011771号