mirror of https://github.com/JDAI-CV/fast-reid.git
19 lines
321 B
Python
19 lines
321 B
Python
# encoding: utf-8
|
|
"""
|
|
@author: xingyu liao
|
|
@contact: sherlockliao01@gmail.com
|
|
"""
|
|
|
|
from fastreid.config import CfgNode as CN
|
|
|
|
|
|
def add_face_cfg(cfg):
|
|
_C = cfg
|
|
|
|
_C.DATASETS.REC_PATH = ""
|
|
|
|
_C.MODEL.BACKBONE.DROPOUT = 0.
|
|
|
|
_C.MODEL.HEADS.PFC = CN({"ENABLED": False})
|
|
_C.MODEL.HEADS.PFC.SAMPLE_RATE = 0.1
|