mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
9684500a57
1. change dataset show to trainset show and testset show seperately 2. add cls layer to easily plug in circle loss and arcface
12 lines
268 B
Python
12 lines
268 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
|