EfficientDet ----torch版本源码解析

EfficientDet 代码模块功能解析

由于官方的代码是tensorflow版本的,我看的源码是zylo大佬写的pytorch版本的代码
地址:EfficientDet-torch版本

由于最近在搞自己的想法,时间不是很充足,所以这次先解析代码中的Backbone脚本,关于数据预处理,loss计算脚本以及后处理脚本下次解析。

  1. 首先介绍EfficientDetBackbone这个类
    具体代码如下:
class EfficientDetBackbone(nn.Module):
    def __init__(self, num_classes=80

你可能感兴趣的:(深度学习,cv,卷积,pytorch,神经网络)