From e606c76ff4de214c2c862e8ca20ac12b4f833b87 Mon Sep 17 00:00:00 2001 From: Pawel Potrykus Date: Thu, 14 Oct 2021 18:49:50 +0200 Subject: [PATCH 1/3] fix remove label annotations --- icevision/core/record_components.py | 1 + 1 file changed, 1 insertion(+) diff --git a/icevision/core/record_components.py b/icevision/core/record_components.py index 7ca92239d..696e9cd44 100644 --- a/icevision/core/record_components.py +++ b/icevision/core/record_components.py @@ -274,6 +274,7 @@ def _autofix(self) -> Dict[str, bool]: def _remove_annotation(self, i): self.label_ids.pop(i) + self.labels.pop(i) def _aggregate_objects(self) -> Dict[str, List[dict]]: return {**super()._aggregate_objects(), "labels": self.label_ids} From 064f551ad8a9cb42bd0005ba0ba2e7ee3eccee33 Mon Sep 17 00:00:00 2001 From: Pawel Potrykus Date: Thu, 14 Oct 2021 18:50:20 +0200 Subject: [PATCH 2/3] add dataset slicing --- icevision/data/dataset.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/icevision/data/dataset.py b/icevision/data/dataset.py index 260b8f9b2..6f67600e1 100644 --- a/icevision/data/dataset.py +++ b/icevision/data/dataset.py @@ -32,13 +32,16 @@ def __len__(self): return len(self.records) def __getitem__(self, i): - record = self.records[i].load() - if self.tfm is not None: - record = self.tfm(record) + if isinstance(i, slice): + return self.__class__(self.records[i], self.tfm) else: - # HACK FIXME - record.set_img(np.array(record.img)) - return record + record = self.records[i].load() + if self.tfm is not None: + record = self.tfm(record) + else: + # HACK FIXME + record.set_img(np.array(record.img)) + return record def __repr__(self): return f"<{self.__class__.__name__} with {len(self.records)} items>" From 500ddf243fae2c12dea53ea83ea98eaef15a3ac1 Mon Sep 17 00:00:00 2001 From: potipot Date: Thu, 11 Aug 2022 17:28:17 +0200 Subject: [PATCH 3/3] add maximum pytorch-lightning version due to deprecation of weights_summary argument passing to Trainer --- setup.cfg | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.cfg b/setup.cfg index 7ee68bac5..28c47d023 100644 --- a/setup.cfg +++ b/setup.cfg @@ -48,7 +48,7 @@ inference = all = fastai >=2.5.2,<2.6 - pytorch-lightning >=1.4.5 + pytorch-lightning >=1.4.5,<1.7.0 wandb >=0.10.7 dev =