目标检测
大雷哥~
这个作者很懒,什么都没留下…
展开
-
通过SSD训练出来的的log日志,利用python绘制loss曲线
利用SSD目标检测算法训练我们的数据集时,将训练结果保存在log文件中。我的log文件如下然后利用python脚本文件将我们需要的数据提取出来,也就是iteration,loss,(或者时epoch,loss).我的脚本文件如下:import matplotlib.pyplot as pltimport numpy as npimport pylab as plfrom mpl_toolkits.axes_grid1.inset_locator import inset_axesline =原创 2021-01-03 12:44:11 · 1969 阅读 · 1 评论 -
pytorch生成的pth文件转成onnx文件再转成ncnn最后真机调试
官方与训练pth文件代码如下import torchimport torchvisionimport torch.nn as nnimport collectionsfrom torchsummary import summaryfrom model import vggfrom torchvision import modelsout_onnx = './vgg16Net.onnx'model = models.vgg16(pretrained=True)params = model原创 2020-12-15 10:49:25 · 2806 阅读 · 0 评论 -
Feature Pyramid Grids论文解读
FPG(Feature Pyramid Grids):特征金字塔网格来了 !性能优于FPN、NAS-FPN等金字塔网络。作者团队:商汤&港中文(陈恺&林达华)&南洋理工大学&FAIR链接:https://arxiv.org/abs/2004.035801.引言特征金字塔网络已在目标检测中被广泛采用,以改进特征表示以更好地处理尺度变化。设计思路在本文...转载 2020-04-15 11:13:19 · 978 阅读 · 0 评论