EfficientNetV2 笔记

AI算法网奇

于 2021-04-06 21:28:18 发布

阅读量2.8k

点赞数

分类专栏：深度学习宝典

本文链接：https://blog.youkuaiyun.com/jacke121/article/details/115471557

版权

深度学习宝典专栏收录该内容

1028 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文详细介绍了EfficientNetV2的网络结构，通过链接提供了PyTorch实现的源代码，帮助读者理解并应用该高效神经网络模型。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch网络结构：

https://github.com/d-li14/efficientnetv2.pytorch/blob/main/effnetv2.py


  
"""
Creates a EfficientNetV2 Model as defined in:
Mingxing Tan, Quoc V. Le. (2021). 
EfficientNetV2: Smaller Models and Faster Training
arXiv preprint arXiv:2104.00298.
import from https://github.com/d-li14/mobilenetv2.pytorch
"""

import torch.nn as nn
import math

__all__ = ['effnetv2_s']


def _make_divisible(v, divisor, min_value=None):
    """
    This function is taken from the original tf repo.
    It ensures that all layers have a channel number that is divisible by 8
    It can be seen