基础概念

ratio:高/宽 h/w

ratio=[0.5, 1.0, 2.0],则表示

scale:各边放大的倍数

scale=[8.],则对应的两条边都放大八倍,即最后的面积变为原来的64倍

获取经过MMDataParallel的模型的module

  • MMDataParallel直接继承torch.nn.parallel.DataParallel,对于model,还需要用.module才能获得对应的model

  • 获取到上面的module之后,直接通过.获取对应的模块就好了,比如:

    runner.model.module.bbox_head # obtain bbox_head module of faster rcnn