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

[tipc] enable tinypose (#8047)

上级 0afb4d5a
......@@ -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
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册