def __init__(self, device):
self.device = device
self.output_feature_sizes = [[Config.input_size[0] // i, Config.input_size[1] // i] for i in Config.yolo_strides]
self.num_classes = Config.num_classes
self.anchor_num_per_level = Config.anchor_num_per_level
self.max_bbox_per_level = Config.max_bbox_per_level
self.strides = torch.tensor(Config.yolo_strides, dtype=torch.float32, device=device)
self.anchors = Config.get_anchors()
self.delta = 0.01
def __init__(self, image_dir, device):
self.device = device
self.num_classes = Config.num_classes
self.strides = Config.yolo_strides
self.anchors = Config.get_anchors().to(device)
self.scale = Config.scale
self.input_size = Config.input_size
self.score_threshold = Config.score_threshold
self.nms_iou_threshold = Config.nms_iou_threshold
self.image, self.image_size = self.__read_image(image_dir)
def __init__(self, device):
self.device = device
self.num_classes = torch.tensor(Config.num_classes, device=device)
self.strides = torch.tensor(Config.yolo_strides, device=device)
self.anchors = Config.get_anchors().to(device)
self.scale = torch.tensor(Config.scale, device=device)
