diff --git a/test_tipc/configs/keypoint/tinypose_128x96.yml b/test_tipc/configs/keypoint/tinypose_128x96.yml index 338d9793c47550fbae8f79c62efca9f62edef2d9..50720e9accdb665db6271a35275be1b7d7849f13 100644 --- a/test_tipc/configs/keypoint/tinypose_128x96.yml +++ b/test_tipc/configs/keypoint/tinypose_128x96.yml @@ -83,25 +83,25 @@ TestDataset: !ImageFolder anno_path: dataset/coco/keypoint_imagelist.txt -worker_num: 2 +worker_num: 16 global_mean: &global_mean [0.485, 0.456, 0.406] global_std: &global_std [0.229, 0.224, 0.225] TrainReader: sample_transforms: - - RandomFlipHalfBodyTransform: - scale: 0.25 - rot: 30 - num_joints_half_body: 8 - prob_half_body: 0.3 - pixel_std: *pixel_std - trainsize: *trainsize - upper_body_ids: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10] - flip_pairs: *flip_perm - - AugmentationbyInformantionDropping: - prob_cutout: 0.5 - offset_factor: 0.05 - num_patch: 1 - trainsize: *trainsize + # - RandomFlipHalfBodyTransform: + # scale: 0.25 + # rot: 30 + # num_joints_half_body: 8 + # prob_half_body: 0.3 + # pixel_std: *pixel_std + # trainsize: *trainsize + # upper_body_ids: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10] + # flip_pairs: *flip_perm + # - AugmentationbyInformantionDropping: + # prob_cutout: 0.5 + # offset_factor: 0.05 + # num_patch: 1 + # trainsize: *trainsize - TopDownAffine: trainsize: *trainsize use_udp: true diff --git a/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt b/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt index 7a61216a596804312f23b41187592c4c71de77b0..9e87cb0c3aeb35e3c60d9896a96fdc3363570aeb 100644 --- a/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt +++ b/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt @@ -49,7 +49,7 @@ inference:./deploy/python/keypoint_infer.py --save_log_path:null --run_benchmark:False null:null -===========================disable_train_benchmark========================== +===========================train_benchmark_params========================== batch_size:512 fp_items:fp32|fp16 epoch:1