
MM
文章平均质量分 53
Abo_luo
竹杖芒鞋轻胜马,一蓑烟雨任平生
展开
-
pytorch使用DCN
pytorch使用DCN前言正文前言关于DCN可形变卷积神经网络相信大家也都不陌生,使用额外的feature map区学习offset,以此达到可形变的效果。感觉和attention比较相似?但是网络实现的代码版本各不相同,编译环境存在很多难以协调等等的问题。而MMopenlab是一个非常不错的工具,其有着实现可形变卷积的方法,因此本文只是做一个引入,如何像正常使卷积一样的使用DCN正文from mmcv.ops import DeformConv2dPackimport torchfrom原创 2022-05-03 11:05:17 · 2518 阅读 · 1 评论 -
mmdetection冻结网络参数训练
MMdetection针对模型backbone进行冻结参数训练问题问题解决可能存在问题总结问题原以为MMdetection针对模型backbone处进行参数冻结训练网上会有大量的教程,刚刚网上查了一下没找到,于是自己动手去发现解决以YOLOX模型为例,调用CSPDarknet时,定义了_freeze_stage方法,并在train方法中调用。self.frozen_stages默认为-1,即不会对模型进行参数冻结,1、可以在定义CspDarknet类时,构造函数__init__中fro原创 2022-04-14 00:36:25 · 5456 阅读 · 12 评论