WALT / mmdet /core /bbox /assigners /base_assigner.py
3v324v23's picture
update demo
a56642d
raw
history blame contribute delete
327 Bytes
from abc import ABCMeta, abstractmethod
class BaseAssigner(metaclass=ABCMeta):
"""Base assigner that assigns boxes to ground truth boxes."""
@abstractmethod
def assign(self, bboxes, gt_bboxes, gt_bboxes_ignore=None, gt_labels=None):
"""Assign boxes to either a ground truth boxes or a negative boxes."""