cvtools.label_convert.coco_to_dets module

class cvtools.label_convert.coco_to_dets.COCO2Dets(anns_file, num_coors=4)[源代码]

基类：object

将DOTA-COCO兼容格式GT转成检测结果表达形式results，保存成pkl results: {

image_id: dets, # image_id必须是anns中有效的id image_id: dets, ...

} dets: {

cls_id：[[位置坐标，得分], [...], ...], cls_id: [[位置坐标，得分], [...], ...], ...

}，

convert(to_file=None)[源代码]
handle_ann(ann)[源代码]

如果想自定义ann处理方式，继承此类，然后重新实现此方法

save_json(to_file)[源代码]
save_pkl(to_file)[源代码]