xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update core.py

This commit is contained in:
Hardik Dava 2023-07-23 13:37:50 +02:00 committed by GitHub
parent ca348401d5
commit 864764e665
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
supervision/dataset/core.py
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):