mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
44cee30dfc
Summary: 1. refactor dataloader and heads 2. bugfix in fastattr, fastclas, fastface and partialreid 3. partial-fc supported in fastface
17 lines
283 B
Python
17 lines
283 B
Python
# encoding: utf-8
|
|
"""
|
|
@author: xingyu liao
|
|
@contact: sherlockliao01@gmail.com
|
|
"""
|
|
|
|
from fastreid.config import CfgNode as CN
|
|
|
|
|
|
def add_attr_config(cfg):
|
|
_C = cfg
|
|
|
|
_C.MODEL.LOSSES.BCE = CN({"WEIGHT_ENABLED": True})
|
|
_C.MODEL.LOSSES.BCE.SCALE = 1.
|
|
|
|
_C.TEST.THRES = 0.5
|