Spaces:
Runtime error
Runtime error
| 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 | |