Skip to content

Commit

Permalink
update configs
Browse files Browse the repository at this point in the history
  • Loading branch information
Ben-Louis committed Mar 17, 2023
1 parent 91509d3 commit ae283e4
Show file tree
Hide file tree
Showing 12 changed files with 414 additions and 245 deletions.
1 change: 0 additions & 1 deletion projects/yolox-pose/configs/_base_

This file was deleted.

1 change: 1 addition & 0 deletions projects/yolox-pose/configs/_base_/datasets
41 changes: 41 additions & 0 deletions projects/yolox-pose/configs/_base_/default_runtime.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
default_scope = 'mmyolo'
custom_imports = dict(imports=['models', 'datasets'])

# hooks
default_hooks = dict(
timer=dict(type='IterTimerHook'),
logger=dict(type='LoggerHook', interval=50),
param_scheduler=dict(type='ParamSchedulerHook'),
checkpoint=dict(type='CheckpointHook', interval=10, max_keep_ckpts=3),
sampler_seed=dict(type='DistSamplerSeedHook'),
visualization=dict(type='mmpose.PoseVisualizationHook', enable=False),
)

# multi-processing backend
env_cfg = dict(
cudnn_benchmark=False,
mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0),
dist_cfg=dict(backend='nccl'),
)

# visualizer
vis_backends = [dict(type='LocalVisBackend')]
visualizer = dict(
type='mmpose.PoseLocalVisualizer',
vis_backends=vis_backends,
name='visualizer')

# logger
log_processor = dict(
type='LogProcessor', window_size=50, by_epoch=True, num_digits=6)
log_level = 'INFO'
load_from = None
resume = False

# file I/O backend
file_client_args = dict(backend='disk')

# training/validation/testing progress
train_cfg = dict()
val_cfg = dict(type='ValLoop')
test_cfg = dict(type='TestLoop')
16 changes: 16 additions & 0 deletions projects/yolox-pose/configs/yolox-pose_l_8xb32-300e_coco.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
_base_ = ['./yolox-pose_s_8xb32-300e_coco.py']

# model settings
model = dict(
init_cfg=dict(checkpoint='https://download.openmmlab.com/mmyolo/v0/yolox/'
'yolox_l_fast__8xb8-300e_coco/yolox_l_fast_8xb8-300e_'
'coco_20230213_160715-c731eb1c.pth'),
backbone=dict(
deepen_factor=1.0,
widen_factor=1.0,
),
neck=dict(
deepen_factor=1.0,
widen_factor=1.0,
),
bbox_head=dict(head_module=dict(widen_factor=1.0)))
16 changes: 16 additions & 0 deletions projects/yolox-pose/configs/yolox-pose_m_8xb32-300e_coco.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
_base_ = ['./yolox-pose_s_8xb32-300e_coco.py']

# model settings
model = dict(
init_cfg=dict(checkpoint='https://download.openmmlab.com/mmyolo/v0/yolox/'
'yolox_m_fast_8xb32-300e-rtmdet-hyp_coco/yolox_m_fast_8xb32'
'-300e-rtmdet-hyp_coco_20230210_144328-e657e182.pth'),
backbone=dict(
deepen_factor=0.33,
widen_factor=0.5,
),
neck=dict(
deepen_factor=0.33,
widen_factor=0.5,
),
bbox_head=dict(head_module=dict(widen_factor=0.5)))

This file was deleted.

125 changes: 0 additions & 125 deletions projects/yolox-pose/configs/yolox-pose_s_8xb32-300e_coco-960.py

This file was deleted.

Loading

0 comments on commit ae283e4

Please sign in to comment.