Pytorch 实践
文章平均质量分 75
玉古路38号
相信自己,一往无前。
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
单阶多层检测器SSD(Single Shot MultiBox Detector)的Pytorch实现
SSD(Single Shot MultiBox Detector) 算法借鉴了Faster RCNN与YOLO的思想,在一阶段网络的基础上使用了固定框进行区域生成,并利用了多层的特征信息,在速度与检测精度上都有了一定的提升。1. SSD算法流程SSD算法流程如上图,输入图像首先经过VGG的基础网络,在此之上又增加了几个卷积层,然后利用3×3的卷积核在6个大小与深浅不同的特征层上进行预测,得到候选框的分类与回归预测值,最后直接预测出结果,或者求得网络的损失。SSD的算法思想,主要可以分为4.原创 2021-06-21 22:06:35 · 395 阅读 · 0 评论 -
【Pytorch Debug】加载保存模型 每次推理结果都不同 / unexpected_keys或者missing_keys / torch.nn.DataParallel( ).cuda()
【bug描述】:模型为仅保存参数的模型,已确定加载语句 “没有问题” ,且每次测试的图片数据固定,但是每次推理得到的结果都不一样。【bug分析】:实际上在加载模型并不是完全没有问题,在使用如下加载语句时出现cnn.load_state_dict(torch.load(os.path.join(MODEL_PATH,"model.pth"))['model'])#保存语句如下:#torch.save({# 'model':cnn.state_dict(),# 'optimiz.原创 2020-10-29 23:02:49 · 2508 阅读 · 0 评论
分享