Skip to content

Commit

Permalink
add new post-process (open-mmlab#288)
Browse files Browse the repository at this point in the history 
* add new post-process

* add new post-process

* add deprecation warnings

* fix the bugs

* changes in topdown.py

* fix some bugs

* simplify code

Signed-off-by: lizz <lizz@sensetime.com>

* change configs accordingly

Signed-off-by: lizz <lizz@sensetime.com>

* One more

Signed-off-by: lizz <lizz@sensetime.com>

* Simplify

Signed-off-by: lizz <lizz@sensetime.com>

Co-authored-by: lizz <lizz@sensetime.com>
  • Loading branch information
@wusize @innerlee
wusize and innerlee authored Nov 30, 2020
1 parent fce9208 commit 6fd6551
Show file tree
Hide file tree
Showing 135 changed files with 206 additions and 295 deletions.
3 changes: 1 addition & 2 deletions configs/hand/resnet/freihand/res50_freihand_224x224.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,9 +54,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/hand/resnet/interhand2d/res50_interhand2d_all_256x256.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,9 +54,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/hand/resnet/interhand2d/res50_interhand2d_human_256x256.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,9 +54,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/hand/resnet/interhand2d/res50_interhand2d_machine_256x256.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,9 +54,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/hand/resnet/onehand10k/res50_onehand10k_256x256.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,9 +54,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/hand/resnet/panoptic/res50_panoptic_256x256.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,9 +54,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/alexnet/coco/alexnet_coco_256x192.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/cpm/coco/cpm_coco_256x192.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -57,9 +57,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/cpm/coco/cpm_coco_384x288.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -57,9 +57,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/cpm/mpii/cpm_mpii_368x368.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -51,9 +51,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco-wholebody/hrnet_w32_coco_wholebody_256x192_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -78,9 +78,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco-wholebody/hrnet_w32_coco_wholebody_384x288_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -78,9 +78,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco-wholebody/hrnet_w48_coco_wholebody_256x192_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -78,9 +78,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco-wholebody/hrnet_w48_coco_wholebody_384x288_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -79,9 +79,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco-wholebody/hrnet_w48_coco_wholebody_384x288_dark_plus.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -79,9 +79,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco/hrnet_w32_coco_256x192_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -80,9 +80,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco/hrnet_w32_coco_384x288_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -80,9 +80,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco/hrnet_w48_coco_256x192_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -80,9 +80,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco/hrnet_w48_coco_384x288_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -80,9 +80,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco/res101_coco_256x192_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco/res101_coco_384x288_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco/res152_coco_256x192_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco/res152_coco_384x288_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco/res50_coco_256x192_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/coco/res50_coco_384x288_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/mpii/hrnet_w32_mpii_256x256_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -74,9 +74,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/darkpose/mpii/hrnet_w48_mpii_256x256_dark.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -74,9 +74,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='unbiased',
shift_heatmap=True,
unbiased_decoding=True,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/hourglass/coco/hourglass52_coco_256x256.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -55,9 +55,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/hourglass/coco/hourglass52_coco_384x384.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -55,9 +55,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/hourglass/mpii/hourglass52_mpii_256x256.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/hourglass/mpii/hourglass52_mpii_384x384.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/hrnet/aic/hrnet_w32_aic_256x192.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -78,9 +78,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/hrnet/aic/hrnet_w32_aic_384x288.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -78,9 +78,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/hrnet/aic/hrnet_w48_aic_256x192.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -78,9 +78,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/hrnet/aic/hrnet_w48_aic_384x288.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -79,9 +79,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
3 changes: 1 addition & 2 deletions configs/top_down/hrnet/coco-wholebody/hrnet_w32_coco_wholebody_256x192.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -78,9 +78,8 @@
train_cfg=dict(),
test_cfg=dict(
flip_test=True,
post_process=True,
post_process='default',
shift_heatmap=True,
unbiased_decoding=False,
modulate_kernel=11),
loss_pose=dict(type='JointsMSELoss', use_target_weight=True))

Expand Down
Loading

0 comments on commit 6fd6551

Please sign in to comment.