liaoxingyu 94c86579a3 fix(heads): fix bug in reduce head
add neck_feat from config, add inplace in leakyrelu for memory save
2020-05-23 10:41:13 +08:00

13 lines
310 B
Python

# encoding: utf-8
"""
@author: liaoxingyu
@contact: sherlockliao01@gmail.com
"""
from .build import REID_HEADS_REGISTRY, build_reid_heads
# import all the meta_arch, so they will be registered
from .linear_head import LinearHead
from .bnneck_head import BNneckHead
from .reduction_head import ReductionHead