目标检测中评估每类目标在IOU为0.5的Precision&&Recall

######################################################################################
###################打印IOU为0.5时的每类bbox大于minsize的Precision&&Recall###############
import json
import numpy as np
from tqdm import tqdm
from collections import defaultdict
from copy import deepcopy

class Dt:
    def __init__(self, img_id, score, bbox, cat):
        """
        single dt
        :param img_id:
        :param score:   a list of score
        :param bbox:  bbox
        """
        self.img_id = img_id
        self.score = score
        self.bbox = bbox
        self.cat = cat
        self.is_match = False


class Gt:
    def __init__(self, img_id, bbox, cat):
        """

        :param img_id:
        :param bbox:
        """
        self.img_id = img_id
        self.bbox = bbox
        self.cat = cat
        self.is_match = False


def compute_iou(bbox1, bbox2):
    x1, y1, w1, h1 = bbox1
    x2, y2, w2, h2 = bbox2
    s_sum = w1 * h1 + w2 * h2

    left = max(x1, x2)
    right = min(x1 + w1, x2 + w2)
    top = max(y1, y2)
    bottom = min(y1 + h1, y2 + h2)

    if left >= right or top >= bottom:
        return 0
    intersect = (right
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值