初始化换发型项目:3个微服务代码 + 部署脚本
包含: - hair_service_sd: 换发型/换发色算法服务 (端口 8801) - photo_service: LoRA 训练调度服务 (端口 32678) - stable-diffusion-webui: SD WebUI 推理服务 (端口 57860) - kohya_ss_home: 训练环境代码 - meidaojia: 监控测试脚本 - setup.sh: 一键部署脚本 (conda环境恢复 + 配置生成 + 完整性检查) - start_all_services.sh: 启动3个服务 - configure.ini.template: 路径模板化 (BASE_DIR自动推导) - conda_envs/py310.yml: py310 环境定义 大文件 (weights/, models/, data/, conda_envs/*.tar.gz 等) 通过 .gitignore 排除, 由网盘单独上传。
This commit is contained in:
@@ -0,0 +1,127 @@
|
||||
AUTO_RESUME: true
|
||||
CUDNN:
|
||||
BENCHMARK: true
|
||||
DETERMINISTIC: false
|
||||
ENABLED: true
|
||||
DATA_DIR: ''
|
||||
GPUS: (0,)
|
||||
OUTPUT_DIR: 'output'
|
||||
LOG_DIR: 'log'
|
||||
WORKERS: 1
|
||||
PRINT_FREQ: 100
|
||||
|
||||
DATASET:
|
||||
COLOR_RGB: true
|
||||
DATASET: 'coco'
|
||||
DATA_FORMAT: jpg
|
||||
FLIP: true
|
||||
NUM_JOINTS_HALF_BODY: 8
|
||||
PROB_HALF_BODY: 0.3
|
||||
ROOT: 'data/coco/'
|
||||
ROT_FACTOR: 45
|
||||
SCALE_FACTOR: 0.35
|
||||
TEST_SET: 'val2017'
|
||||
TRAIN_SET: 'train2017'
|
||||
MODEL:
|
||||
INIT_WEIGHTS: true
|
||||
NAME: pose_hrnet
|
||||
NUM_JOINTS: 25
|
||||
PRETRAINED: 'models/pytorch/imagenet/hrnet_w48-8ef0771d.pth'
|
||||
TARGET_TYPE: gaussian
|
||||
IMAGE_SIZE:
|
||||
- 288
|
||||
- 384
|
||||
HEATMAP_SIZE:
|
||||
- 72
|
||||
- 96
|
||||
SIGMA: 3
|
||||
EXTRA:
|
||||
PRETRAINED_LAYERS:
|
||||
- 'conv1'
|
||||
- 'bn1'
|
||||
- 'conv2'
|
||||
- 'bn2'
|
||||
- 'layer1'
|
||||
- 'transition1'
|
||||
- 'stage2'
|
||||
- 'transition2'
|
||||
- 'stage3'
|
||||
- 'transition3'
|
||||
- 'stage4'
|
||||
FINAL_CONV_KERNEL: 1
|
||||
STAGE2:
|
||||
NUM_MODULES: 1
|
||||
NUM_BRANCHES: 2
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 48
|
||||
- 96
|
||||
FUSE_METHOD: SUM
|
||||
STAGE3:
|
||||
NUM_MODULES: 4
|
||||
NUM_BRANCHES: 3
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 48
|
||||
- 96
|
||||
- 192
|
||||
FUSE_METHOD: SUM
|
||||
STAGE4:
|
||||
NUM_MODULES: 3
|
||||
NUM_BRANCHES: 4
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 48
|
||||
- 96
|
||||
- 192
|
||||
- 384
|
||||
FUSE_METHOD: SUM
|
||||
LOSS:
|
||||
USE_TARGET_WEIGHT: true
|
||||
TRAIN:
|
||||
BATCH_SIZE_PER_GPU: 24
|
||||
SHUFFLE: true
|
||||
BEGIN_EPOCH: 0
|
||||
END_EPOCH: 210
|
||||
OPTIMIZER: adam
|
||||
LR: 0.001
|
||||
LR_FACTOR: 0.1
|
||||
LR_STEP:
|
||||
- 170
|
||||
- 200
|
||||
WD: 0.0001
|
||||
GAMMA1: 0.99
|
||||
GAMMA2: 0.0
|
||||
MOMENTUM: 0.9
|
||||
NESTEROV: false
|
||||
TEST:
|
||||
BATCH_SIZE_PER_GPU: 1
|
||||
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json'
|
||||
BBOX_THRE: 1.0
|
||||
IMAGE_THRE: 0.0
|
||||
IN_VIS_THRE: 0.2
|
||||
MODEL_FILE: ''
|
||||
NMS_THRE: 1.0
|
||||
OKS_THRE: 0.9
|
||||
USE_GT_BBOX: true
|
||||
FLIP_TEST: false
|
||||
POST_PROCESS: true
|
||||
SHIFT_HEATMAP: true
|
||||
DEBUG:
|
||||
DEBUG: true
|
||||
SAVE_BATCH_IMAGES_GT: true
|
||||
SAVE_BATCH_IMAGES_PRED: true
|
||||
SAVE_HEATMAPS_GT: true
|
||||
SAVE_HEATMAPS_PRED: true
|
||||
@@ -0,0 +1,127 @@
|
||||
AUTO_RESUME: true
|
||||
CUDNN:
|
||||
BENCHMARK: true
|
||||
DETERMINISTIC: false
|
||||
ENABLED: true
|
||||
DATA_DIR: ''
|
||||
GPUS: (0,1,2,3)
|
||||
OUTPUT_DIR: 'output'
|
||||
LOG_DIR: 'log'
|
||||
WORKERS: 24
|
||||
PRINT_FREQ: 100
|
||||
|
||||
DATASET:
|
||||
COLOR_RGB: true
|
||||
DATASET: 'coco'
|
||||
DATA_FORMAT: jpg
|
||||
FLIP: true
|
||||
NUM_JOINTS_HALF_BODY: 8
|
||||
PROB_HALF_BODY: 0.3
|
||||
ROOT: 'data/coco/'
|
||||
ROT_FACTOR: 45
|
||||
SCALE_FACTOR: 0.35
|
||||
TEST_SET: 'val2017'
|
||||
TRAIN_SET: 'train2017'
|
||||
MODEL:
|
||||
INIT_WEIGHTS: true
|
||||
NAME: pose_hrnet
|
||||
NUM_JOINTS: 17
|
||||
PRETRAINED: 'models/pytorch/imagenet/hrnet_w32-36af842e.pth'
|
||||
TARGET_TYPE: gaussian
|
||||
IMAGE_SIZE:
|
||||
- 192
|
||||
- 256
|
||||
HEATMAP_SIZE:
|
||||
- 48
|
||||
- 64
|
||||
SIGMA: 2
|
||||
EXTRA:
|
||||
PRETRAINED_LAYERS:
|
||||
- 'conv1'
|
||||
- 'bn1'
|
||||
- 'conv2'
|
||||
- 'bn2'
|
||||
- 'layer1'
|
||||
- 'transition1'
|
||||
- 'stage2'
|
||||
- 'transition2'
|
||||
- 'stage3'
|
||||
- 'transition3'
|
||||
- 'stage4'
|
||||
FINAL_CONV_KERNEL: 1
|
||||
STAGE2:
|
||||
NUM_MODULES: 1
|
||||
NUM_BRANCHES: 2
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 32
|
||||
- 64
|
||||
FUSE_METHOD: SUM
|
||||
STAGE3:
|
||||
NUM_MODULES: 4
|
||||
NUM_BRANCHES: 3
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 32
|
||||
- 64
|
||||
- 128
|
||||
FUSE_METHOD: SUM
|
||||
STAGE4:
|
||||
NUM_MODULES: 3
|
||||
NUM_BRANCHES: 4
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 32
|
||||
- 64
|
||||
- 128
|
||||
- 256
|
||||
FUSE_METHOD: SUM
|
||||
LOSS:
|
||||
USE_TARGET_WEIGHT: true
|
||||
TRAIN:
|
||||
BATCH_SIZE_PER_GPU: 32
|
||||
SHUFFLE: true
|
||||
BEGIN_EPOCH: 0
|
||||
END_EPOCH: 210
|
||||
OPTIMIZER: adam
|
||||
LR: 0.001
|
||||
LR_FACTOR: 0.1
|
||||
LR_STEP:
|
||||
- 170
|
||||
- 200
|
||||
WD: 0.0001
|
||||
GAMMA1: 0.99
|
||||
GAMMA2: 0.0
|
||||
MOMENTUM: 0.9
|
||||
NESTEROV: false
|
||||
TEST:
|
||||
BATCH_SIZE_PER_GPU: 32
|
||||
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json'
|
||||
BBOX_THRE: 1.0
|
||||
IMAGE_THRE: 0.0
|
||||
IN_VIS_THRE: 0.2
|
||||
MODEL_FILE: ''
|
||||
NMS_THRE: 1.0
|
||||
OKS_THRE: 0.9
|
||||
USE_GT_BBOX: true
|
||||
FLIP_TEST: true
|
||||
POST_PROCESS: true
|
||||
SHIFT_HEATMAP: true
|
||||
DEBUG:
|
||||
DEBUG: true
|
||||
SAVE_BATCH_IMAGES_GT: true
|
||||
SAVE_BATCH_IMAGES_PRED: true
|
||||
SAVE_HEATMAPS_GT: true
|
||||
SAVE_HEATMAPS_PRED: true
|
||||
@@ -0,0 +1,127 @@
|
||||
AUTO_RESUME: true
|
||||
CUDNN:
|
||||
BENCHMARK: true
|
||||
DETERMINISTIC: false
|
||||
ENABLED: true
|
||||
DATA_DIR: ''
|
||||
GPUS: (0,1,2,3)
|
||||
OUTPUT_DIR: 'output'
|
||||
LOG_DIR: 'log'
|
||||
WORKERS: 24
|
||||
PRINT_FREQ: 100
|
||||
|
||||
DATASET:
|
||||
COLOR_RGB: true
|
||||
DATASET: 'coco'
|
||||
DATA_FORMAT: jpg
|
||||
FLIP: true
|
||||
NUM_JOINTS_HALF_BODY: 8
|
||||
PROB_HALF_BODY: 0.3
|
||||
ROOT: 'data/coco/'
|
||||
ROT_FACTOR: 45
|
||||
SCALE_FACTOR: 0.35
|
||||
TEST_SET: 'val2017'
|
||||
TRAIN_SET: 'train2017'
|
||||
MODEL:
|
||||
INIT_WEIGHTS: true
|
||||
NAME: pose_hrnet
|
||||
NUM_JOINTS: 17
|
||||
PRETRAINED: 'models/pytorch/imagenet/hrnet_w32-36af842e.pth'
|
||||
TARGET_TYPE: gaussian
|
||||
IMAGE_SIZE:
|
||||
- 288
|
||||
- 384
|
||||
HEATMAP_SIZE:
|
||||
- 72
|
||||
- 96
|
||||
SIGMA: 3
|
||||
EXTRA:
|
||||
PRETRAINED_LAYERS:
|
||||
- 'conv1'
|
||||
- 'bn1'
|
||||
- 'conv2'
|
||||
- 'bn2'
|
||||
- 'layer1'
|
||||
- 'transition1'
|
||||
- 'stage2'
|
||||
- 'transition2'
|
||||
- 'stage3'
|
||||
- 'transition3'
|
||||
- 'stage4'
|
||||
FINAL_CONV_KERNEL: 1
|
||||
STAGE2:
|
||||
NUM_MODULES: 1
|
||||
NUM_BRANCHES: 2
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 32
|
||||
- 64
|
||||
FUSE_METHOD: SUM
|
||||
STAGE3:
|
||||
NUM_MODULES: 4
|
||||
NUM_BRANCHES: 3
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 32
|
||||
- 64
|
||||
- 128
|
||||
FUSE_METHOD: SUM
|
||||
STAGE4:
|
||||
NUM_MODULES: 3
|
||||
NUM_BRANCHES: 4
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 32
|
||||
- 64
|
||||
- 128
|
||||
- 256
|
||||
FUSE_METHOD: SUM
|
||||
LOSS:
|
||||
USE_TARGET_WEIGHT: true
|
||||
TRAIN:
|
||||
BATCH_SIZE_PER_GPU: 32
|
||||
SHUFFLE: true
|
||||
BEGIN_EPOCH: 0
|
||||
END_EPOCH: 210
|
||||
OPTIMIZER: adam
|
||||
LR: 0.001
|
||||
LR_FACTOR: 0.1
|
||||
LR_STEP:
|
||||
- 170
|
||||
- 200
|
||||
WD: 0.0001
|
||||
GAMMA1: 0.99
|
||||
GAMMA2: 0.0
|
||||
MOMENTUM: 0.9
|
||||
NESTEROV: false
|
||||
TEST:
|
||||
BATCH_SIZE_PER_GPU: 32
|
||||
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json'
|
||||
BBOX_THRE: 1.0
|
||||
IMAGE_THRE: 0.0
|
||||
IN_VIS_THRE: 0.2
|
||||
MODEL_FILE: ''
|
||||
NMS_THRE: 1.0
|
||||
OKS_THRE: 0.9
|
||||
USE_GT_BBOX: true
|
||||
FLIP_TEST: true
|
||||
POST_PROCESS: true
|
||||
SHIFT_HEATMAP: true
|
||||
DEBUG:
|
||||
DEBUG: true
|
||||
SAVE_BATCH_IMAGES_GT: true
|
||||
SAVE_BATCH_IMAGES_PRED: true
|
||||
SAVE_HEATMAPS_GT: true
|
||||
SAVE_HEATMAPS_PRED: true
|
||||
@@ -0,0 +1,127 @@
|
||||
AUTO_RESUME: true
|
||||
CUDNN:
|
||||
BENCHMARK: true
|
||||
DETERMINISTIC: false
|
||||
ENABLED: true
|
||||
DATA_DIR: ''
|
||||
GPUS: (0,1,2,3)
|
||||
OUTPUT_DIR: 'output'
|
||||
LOG_DIR: 'log'
|
||||
WORKERS: 24
|
||||
PRINT_FREQ: 100
|
||||
|
||||
DATASET:
|
||||
COLOR_RGB: true
|
||||
DATASET: 'coco'
|
||||
DATA_FORMAT: jpg
|
||||
FLIP: true
|
||||
NUM_JOINTS_HALF_BODY: 8
|
||||
PROB_HALF_BODY: 0.3
|
||||
ROOT: 'data/coco/'
|
||||
ROT_FACTOR: 45
|
||||
SCALE_FACTOR: 0.35
|
||||
TEST_SET: 'val2017'
|
||||
TRAIN_SET: 'train2017'
|
||||
MODEL:
|
||||
INIT_WEIGHTS: true
|
||||
NAME: pose_hrnet
|
||||
NUM_JOINTS: 17
|
||||
PRETRAINED: 'models/pytorch/imagenet/hrnet_w48-8ef0771d.pth'
|
||||
TARGET_TYPE: gaussian
|
||||
IMAGE_SIZE:
|
||||
- 192
|
||||
- 256
|
||||
HEATMAP_SIZE:
|
||||
- 48
|
||||
- 64
|
||||
SIGMA: 2
|
||||
EXTRA:
|
||||
PRETRAINED_LAYERS:
|
||||
- 'conv1'
|
||||
- 'bn1'
|
||||
- 'conv2'
|
||||
- 'bn2'
|
||||
- 'layer1'
|
||||
- 'transition1'
|
||||
- 'stage2'
|
||||
- 'transition2'
|
||||
- 'stage3'
|
||||
- 'transition3'
|
||||
- 'stage4'
|
||||
FINAL_CONV_KERNEL: 1
|
||||
STAGE2:
|
||||
NUM_MODULES: 1
|
||||
NUM_BRANCHES: 2
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 48
|
||||
- 96
|
||||
FUSE_METHOD: SUM
|
||||
STAGE3:
|
||||
NUM_MODULES: 4
|
||||
NUM_BRANCHES: 3
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 48
|
||||
- 96
|
||||
- 192
|
||||
FUSE_METHOD: SUM
|
||||
STAGE4:
|
||||
NUM_MODULES: 3
|
||||
NUM_BRANCHES: 4
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 48
|
||||
- 96
|
||||
- 192
|
||||
- 384
|
||||
FUSE_METHOD: SUM
|
||||
LOSS:
|
||||
USE_TARGET_WEIGHT: true
|
||||
TRAIN:
|
||||
BATCH_SIZE_PER_GPU: 32
|
||||
SHUFFLE: true
|
||||
BEGIN_EPOCH: 0
|
||||
END_EPOCH: 210
|
||||
OPTIMIZER: adam
|
||||
LR: 0.001
|
||||
LR_FACTOR: 0.1
|
||||
LR_STEP:
|
||||
- 170
|
||||
- 200
|
||||
WD: 0.0001
|
||||
GAMMA1: 0.99
|
||||
GAMMA2: 0.0
|
||||
MOMENTUM: 0.9
|
||||
NESTEROV: false
|
||||
TEST:
|
||||
BATCH_SIZE_PER_GPU: 32
|
||||
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json'
|
||||
BBOX_THRE: 1.0
|
||||
IMAGE_THRE: 0.0
|
||||
IN_VIS_THRE: 0.2
|
||||
MODEL_FILE: ''
|
||||
NMS_THRE: 1.0
|
||||
OKS_THRE: 0.9
|
||||
USE_GT_BBOX: true
|
||||
FLIP_TEST: true
|
||||
POST_PROCESS: true
|
||||
SHIFT_HEATMAP: true
|
||||
DEBUG:
|
||||
DEBUG: true
|
||||
SAVE_BATCH_IMAGES_GT: true
|
||||
SAVE_BATCH_IMAGES_PRED: true
|
||||
SAVE_HEATMAPS_GT: true
|
||||
SAVE_HEATMAPS_PRED: true
|
||||
@@ -0,0 +1,127 @@
|
||||
AUTO_RESUME: true
|
||||
CUDNN:
|
||||
BENCHMARK: true
|
||||
DETERMINISTIC: false
|
||||
ENABLED: true
|
||||
DATA_DIR: ''
|
||||
GPUS: (0,)
|
||||
OUTPUT_DIR: 'output'
|
||||
LOG_DIR: 'log'
|
||||
WORKERS: 1
|
||||
PRINT_FREQ: 100
|
||||
|
||||
DATASET:
|
||||
COLOR_RGB: true
|
||||
DATASET: 'coco'
|
||||
DATA_FORMAT: jpg
|
||||
FLIP: true
|
||||
NUM_JOINTS_HALF_BODY: 8
|
||||
PROB_HALF_BODY: 0.3
|
||||
ROOT: 'data/coco/'
|
||||
ROT_FACTOR: 45
|
||||
SCALE_FACTOR: 0.35
|
||||
TEST_SET: 'val2017'
|
||||
TRAIN_SET: 'train2017'
|
||||
MODEL:
|
||||
INIT_WEIGHTS: true
|
||||
NAME: pose_hrnet
|
||||
NUM_JOINTS: 17
|
||||
PRETRAINED: 'models/pytorch/imagenet/hrnet_w48-8ef0771d.pth'
|
||||
TARGET_TYPE: gaussian
|
||||
IMAGE_SIZE:
|
||||
- 288
|
||||
- 384
|
||||
HEATMAP_SIZE:
|
||||
- 72
|
||||
- 96
|
||||
SIGMA: 3
|
||||
EXTRA:
|
||||
PRETRAINED_LAYERS:
|
||||
- 'conv1'
|
||||
- 'bn1'
|
||||
- 'conv2'
|
||||
- 'bn2'
|
||||
- 'layer1'
|
||||
- 'transition1'
|
||||
- 'stage2'
|
||||
- 'transition2'
|
||||
- 'stage3'
|
||||
- 'transition3'
|
||||
- 'stage4'
|
||||
FINAL_CONV_KERNEL: 1
|
||||
STAGE2:
|
||||
NUM_MODULES: 1
|
||||
NUM_BRANCHES: 2
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 48
|
||||
- 96
|
||||
FUSE_METHOD: SUM
|
||||
STAGE3:
|
||||
NUM_MODULES: 4
|
||||
NUM_BRANCHES: 3
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 48
|
||||
- 96
|
||||
- 192
|
||||
FUSE_METHOD: SUM
|
||||
STAGE4:
|
||||
NUM_MODULES: 3
|
||||
NUM_BRANCHES: 4
|
||||
BLOCK: BASIC
|
||||
NUM_BLOCKS:
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
- 4
|
||||
NUM_CHANNELS:
|
||||
- 48
|
||||
- 96
|
||||
- 192
|
||||
- 384
|
||||
FUSE_METHOD: SUM
|
||||
LOSS:
|
||||
USE_TARGET_WEIGHT: true
|
||||
TRAIN:
|
||||
BATCH_SIZE_PER_GPU: 24
|
||||
SHUFFLE: true
|
||||
BEGIN_EPOCH: 0
|
||||
END_EPOCH: 210
|
||||
OPTIMIZER: adam
|
||||
LR: 0.001
|
||||
LR_FACTOR: 0.1
|
||||
LR_STEP:
|
||||
- 170
|
||||
- 200
|
||||
WD: 0.0001
|
||||
GAMMA1: 0.99
|
||||
GAMMA2: 0.0
|
||||
MOMENTUM: 0.9
|
||||
NESTEROV: false
|
||||
TEST:
|
||||
BATCH_SIZE_PER_GPU: 1
|
||||
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json'
|
||||
BBOX_THRE: 1.0
|
||||
IMAGE_THRE: 0.0
|
||||
IN_VIS_THRE: 0.2
|
||||
MODEL_FILE: ''
|
||||
NMS_THRE: 1.0
|
||||
OKS_THRE: 0.9
|
||||
USE_GT_BBOX: true
|
||||
FLIP_TEST: false
|
||||
POST_PROCESS: true
|
||||
SHIFT_HEATMAP: true
|
||||
DEBUG:
|
||||
DEBUG: true
|
||||
SAVE_BATCH_IMAGES_GT: true
|
||||
SAVE_BATCH_IMAGES_PRED: true
|
||||
SAVE_HEATMAPS_GT: true
|
||||
SAVE_HEATMAPS_PRED: true
|
||||
Reference in New Issue
Block a user