easy-Fpn源码解读(三)：bbox

easy-Fpn源码解读(二)：bbox

bbox代码解析

import torch
from torch import Tensor


class BBox(object):
    # 初始化方法：四个坐标点
    def __init__(self, left: float, top: float, right: float, bottom: float):
        super().__init__()
        self.left = left
        self.top = top
        self.right = right
        self.bottom = bottom
     
    # 输出实例化对象的坐标信息
    def __repr__(self) -> str:
        return 'BBox[l={:.1f}, t={:.1f}, r={:.1f}, b={:.1f}]'.format(
            self.left, self.top, self.right, self.bottom)
    
    # 将bbox转换成列表
    def tolist(self):
        return [self.left, self.top, self.right, self.bottom]

    @staticmethod
    # 类调用，静态方法：把(x0,y0,x1,y1)转换成(x,y,w,h)
    def to_center_base(bboxes: Tensor):
        return torch.stack([
            (bboxes[:, 0] + bboxes[:, 2]) / 2,
            (bboxes[:, 1] + bboxes[:, 3]) / 2,
            bboxes[:, 2] - bboxes[:, 0],
            bboxes[:, 3] - bboxes[:, 1]
        ], dim=1)

    @staticmethod
    # 把(x,y,w,h)转换成(x0,y0,x1,y1)
    def from_center_base(center_based_bboxes: Tensor) -> Tensor:
        return torch.stack([
            center_based_bboxes[:, 0] - center_based_bboxes[:, 2] / 2,
            center_based_bboxes[:, 1] - center_based_bboxes[:, 3] / 2,
            center_based_bboxes[:, 0] + center_based_bboxes[:,