mirror of https://github.com/JDAI-CV/fast-reid.git
25 lines
602 B
Python
25 lines
602 B
Python
# encoding: utf-8
|
|
"""
|
|
@author: liaoxingyu
|
|
@contact: sherlockliao01@gmail.com
|
|
"""
|
|
from torch import nn
|
|
|
|
from .batch_drop import BatchDrop
|
|
from .attention import *
|
|
from .batch_norm import *
|
|
from .context_block import ContextBlock
|
|
from .non_local import Non_local
|
|
from .se_layer import SELayer
|
|
from .frn import FRN, TLU
|
|
from .activation import *
|
|
from .gem_pool import GeneralizedMeanPoolingP, AdaptiveAvgMaxPool2d
|
|
from .arcface import Arcface
|
|
from .circle import Circle
|
|
from .splat import SplAtConv2d
|
|
|
|
|
|
class Flatten(nn.Module):
|
|
def forward(self, input):
|
|
return input.view(input.size(0), -1)
|