pytorch
文章平均质量分 67
AManFromEarth
这个作者很懒,什么都没留下…
展开
-
目标检测-论文阅读-FPN(Feature Pyramid Networks for Object Detection)
FPN论文地址,以及pytorch版本的github地址,实际上最新版本的pytorch已经集成了FPN,所以建议去看该代码。立意FPN的立意很直接,如图1所示的目标检测常用的abc三种特征提取结构:(a) 将图像依次缩放成金字塔样式并分别提取特征,优点是能提取多个尺度的带有较强语义信息的特征表示,缺点是预测时间是之前的多倍、训练较占显存;(b) Faster RCNN用的无金字塔结构,缺点和优点与(a)相反;© SSD使用的网络内特征金字塔层次,优点是能产生不同分辨率的特征图,缺点是不同的深度(原创 2020-06-17 20:01:08 · 1032 阅读 · 1 评论 -
使用pytorch读取、使用预训练模型进行finetune:以Resnet-101为例
在使用pytorch进行网络训练的时候,有时候不可避免的使用迁移学习(trainsfer learning),即使用已经训练好的模型(如resnet、inception等),固定其已经训练好的网络层参数,然后进行finetune。 以下代码是以resnet-101为例使用pytorch进行finetune的操作:#导入必要模块import torchimport torch.nn as ...原创 2018-07-16 21:49:34 · 22877 阅读 · 0 评论