未验证 提交 8f37ad0f 编写于 作者: L LokeZhou 提交者: GitHub

[tipc] enable tinypose (#8047)

上级 0afb4d5a
...@@ -83,25 +83,25 @@ TestDataset: ...@@ -83,25 +83,25 @@ TestDataset:
!ImageFolder !ImageFolder
anno_path: dataset/coco/keypoint_imagelist.txt anno_path: dataset/coco/keypoint_imagelist.txt
worker_num: 2 worker_num: 16
global_mean: &global_mean [0.485, 0.456, 0.406] global_mean: &global_mean [0.485, 0.456, 0.406]
global_std: &global_std [0.229, 0.224, 0.225] global_std: &global_std [0.229, 0.224, 0.225]
TrainReader: TrainReader:
sample_transforms: sample_transforms:
- RandomFlipHalfBodyTransform: # - RandomFlipHalfBodyTransform:
scale: 0.25 # scale: 0.25
rot: 30 # rot: 30
num_joints_half_body: 8 # num_joints_half_body: 8
prob_half_body: 0.3 # prob_half_body: 0.3
pixel_std: *pixel_std # pixel_std: *pixel_std
trainsize: *trainsize # trainsize: *trainsize
upper_body_ids: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10] # upper_body_ids: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
flip_pairs: *flip_perm # flip_pairs: *flip_perm
- AugmentationbyInformantionDropping: # - AugmentationbyInformantionDropping:
prob_cutout: 0.5 # prob_cutout: 0.5
offset_factor: 0.05 # offset_factor: 0.05
num_patch: 1 # num_patch: 1
trainsize: *trainsize # trainsize: *trainsize
- TopDownAffine: - TopDownAffine:
trainsize: *trainsize trainsize: *trainsize
use_udp: true use_udp: true
......
...@@ -49,7 +49,7 @@ inference:./deploy/python/keypoint_infer.py ...@@ -49,7 +49,7 @@ inference:./deploy/python/keypoint_infer.py
--save_log_path:null --save_log_path:null
--run_benchmark:False --run_benchmark:False
null:null null:null
===========================disable_train_benchmark========================== ===========================train_benchmark_params==========================
batch_size:512 batch_size:512
fp_items:fp32|fp16 fp_items:fp32|fp16
epoch:1 epoch:1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册