mirror of https://github.com/menyifang/DCT-Net
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
624 B
Python
24 lines
624 B
Python
3 years ago
|
import os
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
from easydict import EasyDict as edict
|
||
|
|
|
||
|
|
config = edict()
|
||
|
|
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
|
||
|
|
|
||
|
|
config.DETECT = edict()
|
||
|
|
config.DETECT.topk = 10
|
||
|
|
config.DETECT.thres = 0.8
|
||
|
|
config.DETECT.input_shape = (512, 512, 3)
|
||
|
|
config.KEYPOINTS = edict()
|
||
|
|
config.KEYPOINTS.p_num = 68
|
||
|
|
config.KEYPOINTS.base_extend_range = [0.2, 0.3]
|
||
|
|
config.KEYPOINTS.input_shape = (160, 160, 3)
|
||
|
|
config.TRACE = edict()
|
||
|
|
config.TRACE.pixel_thres = 1
|
||
|
|
config.TRACE.smooth_box = 0.3
|
||
|
|
config.TRACE.smooth_landmark = 0.95
|
||
|
|
config.TRACE.iou_thres = 0.5
|
||
|
|
config.DATA = edict()
|
||
|
|
config.DATA.pixel_means = np.array([123., 116., 103.]) # RGB
|