2020年12月17日

Pytorch_模型转Caffe（二）解析Pytorch模型*.pth

摘要： Pytorch_模型转Caffe（二）解析Pytorch模型*.pth 1. Pytorch模型保存于读取 a. 保存、加载权重 # 模型保存（仅保存权重） torch.save(model_object.state_dict(), './weights.pth') # 模型加载（先创建模型，、再导 阅读全文

posted @ 2020-12-17 15:17 wangxiaobei2019 阅读(988) 评论(0) 推荐(0) 编辑

Pytorch_模型转Caffe（一）解析caffemodel与prototxt

摘要： #Pytorch_模型转Caffe（一） 1.Caffe简介 2.Caffe进行目标检测任务 利用ssd进行目标检测任务，主要步骤如下（重点是模型的移植） 3.Caffe五大组件 4.caffemodel 包含了prototxt(除了solver.prototxt) 和 weights bias p 阅读全文

posted @ 2020-12-17 09:41 wangxiaobei2019 阅读(3116) 评论(0) 推荐(1) 编辑

导航