retinanet_r50_fpn_1x_widerface.py 342 B

12345678910
  1. _base_ = [
  2. '../_base_/models/retinanet_r50_fpn.py',
  3. '../_base_/datasets/wider_face.py', '../_base_/schedules/schedule_1x.py',
  4. '../_base_/default_runtime.py'
  5. ]
  6. # model settings
  7. model = dict(bbox_head=dict(num_classes=1))
  8. # optimizer
  9. optim_wrapper = dict(
  10. optimizer=dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001))