空间分组卷积

      from collections import OrderedDict
      import time
      import torch
      import torch.nn as nn
      import torch.nn.functional as F
      from torch.nn import init
      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 here:
       https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet.py
       :param v:
       :param divisor:
       :param min_value:
       :return:
       """
      if min_value is None:
       min_value = divisor
       new_v = max(min_value, int(v + divisor / 2) // divisor * divisor)
      # Make sure that round down does not go down by more than 10%.
      if new_v < 0.9 * v:
       new_v += divisor
      return new_v
      class LinearBottleneck(nn.Module):
      def __init__(self, inplanes, outplanes, stride=1, t=6, activation=
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/106952943