mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
890224f25c
Summary: support classification and refactor build_dataloader which can support explicit parameters passing
12 lines
265 B
Python
12 lines
265 B
Python
# encoding: utf-8
|
|
"""
|
|
@author: liaoxingyu
|
|
@contact: sherlockliao01@gmail.com
|
|
"""
|
|
|
|
from .build import REID_HEADS_REGISTRY, build_heads
|
|
|
|
# import all the meta_arch, so they will be registered
|
|
from .embedding_head import EmbeddingHead
|
|
from .clas_head import ClasHead
|