Created by: lxastro
Add iou aware module (see https://arxiv.org/abs/1912.05992). Major changes include: 1) add IoU aware loss. 2) add an IoU predict branch in yolo_head. Imporving mAP on COCO from 43.2 to 43.6.
Created by: lxastro
Add iou aware module (see https://arxiv.org/abs/1912.05992). Major changes include: 1) add IoU aware loss. 2) add an IoU predict branch in yolo_head. Imporving mAP on COCO from 43.2 to 43.6.