aee97b14f7b18ed505393642083cfb24b54c9a8b,lib/core/test_retinanet.py,,multi_gpu_test_retinanet_on_dataset,#Any#Any#Any#,258

Before Change

    // Combine the results from each subprocess now
    boxes, scores, classes, image_ids = [], [], [], []
    for det_data in outputs:
        boxes.extend(det_data["boxes"])
        scores.extend(det_data["scores"])
        classes.extend(det_data["classes"])
        image_ids.extend(det_data["ids"])
    return boxes, scores, classes, image_ids,

After Change

    all_boxes = [[] for _ in range(cfg.MODEL.NUM_CLASSES)]
    for det_data in outputs:
        all_boxes_batch = det_data["all_boxes"]
        for cls_idx in range(1, cfg.MODEL.NUM_CLASSES):
            all_boxes[cls_idx] += all_boxes_batch[cls_idx]

    // Save the computed detections
    det_file = os.path.join(output_dir, "detections.pkl")
    cfg_yaml = yaml.dump(cfg)
    save_object(
        dict(all_boxes=all_boxes, cfg=cfg_yaml),

Instances