Update core.py

This commit is contained in:
Hardik Dava authored and GitHub committed 2023-07-23 13:37:50 +02:00
1 parent ca348401d5
commit 864764e665
1 file changed
-20
-20
View File
@@ -46,14 +46,6 @@ class BaseDataset(ABC):
) -> Tuple[BaseDataset, BaseDataset]:
pass
def add_instance(
self, filename: str, image: np.ndarray, detections: Detections
) -> None:
pass
def add_class_names(self, class_names: List[str]) -> None:
pass
@dataclass
class DetectionDataset(BaseDataset):
@@ -511,18 +503,6 @@ class DetectionDataset(BaseDataset):
classes=merged_classes, images=merged_images, annotations=merged_annotations
)
def add_class_names(self, class_names: List[str]):
self.classes = class_names
def add_instance(self, filename: str, image: np.ndarray, detections: Detections):
if filename not in self.annotations.keys():
self.images[filename] = image
self.annotations[filename] = detections
@classmethod
def emty(cls):
return cls(classes=[], images={}, annotations={})
@dataclass
class ClassificationDataset(BaseDataset):