EfficientDet 代码模块功能解析
由于官方的代码是tensorflow版本的,我看的源码是zylo大佬写的pytorch版本的代码
地址:EfficientDet-torch版本
由于最近在搞自己的想法,时间不是很充足,所以这次先解析代码中的Backbone脚本,关于数据预处理,loss计算脚本以及后处理脚本下次解析。
- 首先介绍EfficientDetBackbone这个类
具体代码如下:
class EfficientDetBackbone(nn.Module):
def