mmdetection
文章平均质量分 62
K . U . I
个人博客:http://www.xuyihao.top/
知乎:https://www.zhihu.com/people/a-fu-jia-de-luo-1-86
展开
-
[源码分析]rotate_roi_align反向过程
前言之前已经梳理过了前向传播过程,接下来要介绍的是反向传播过程的源码解析。正文firstat::Tensor ROIAlignRotated_backward_cpu( const at::Tensor& grad, const at::Tensor& rois, const float spatial_scale, const int pooled_height, const int pooled_width, const int batch_size, const原创 2021-09-20 11:49:07 · 584 阅读 · 0 评论 -
[源码分析]rotate_roi_align前向过程
前言最近在忙于学业的同时,终于将之前一直心心念念的rotate roi align的源码通读了一遍,通读完毕之后感觉真的收获颇丰。由于想限制篇幅,不给读者造成太多的阅读压力,所以本篇博客只介绍前向过程的源码,下篇博客再介绍反向过程的源码。(ps:本次阅读的源码主要参考自detectron2框架,mmdetection框架的源码也是基于此进行了些许的优化,etc:加入了旋转方向的判断)。正文通常提出的算子在pytorch本身不支持自动求导的情况下,都会利用torch.autograd.Function原创 2021-09-19 23:55:11 · 1140 阅读 · 0 评论 -
mmdetection中assign和sample结构的简要解析
前言最近正在学习mmdetection中assign和sample的相关结构,故写此篇博客做一些简单的介绍。正文assign在mmdet中,官方提供了许多种anchor与bbox的匹配机制,其中包括:max_iou_assigner,atss_assigner等等。但是不论这些匹配细则发生怎样的变化,但是有一条语句却始终没有变化:return AssignResult(num_gt, assigned_gt_inds, max_overlaps, labels=assigned_labels)#原创 2021-09-05 16:15:11 · 1531 阅读 · 2 评论