deepvisiontools 0.0.2__tar.gz → 0.0.3__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (61) hide show
  1. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/PKG-INFO +1 -1
  2. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/__init__.py +1 -1
  3. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/formats/base_data.py +4 -1
  4. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools.egg-info/PKG-INFO +1 -1
  5. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/LICENSE.txt +0 -0
  6. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/README.md +0 -0
  7. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/pyproject.toml +0 -0
  8. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/setup.cfg +0 -0
  9. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/setup.py +0 -0
  10. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/config/__init__.py +0 -0
  11. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/config/config.py +0 -0
  12. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/config/errors.py +0 -0
  13. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/data/__init__.py +0 -0
  14. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/data/additional_augmentations.py +0 -0
  15. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/data/augmentation_class.py +0 -0
  16. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/data/batch_augmentations.py +0 -0
  17. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/data/data_reader/__init__.py +0 -0
  18. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/data/data_reader/readers.py +0 -0
  19. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/data/dataset.py +0 -0
  20. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/data/errors.py +0 -0
  21. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/formats/__init__.py +0 -0
  22. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/formats/base_formats.py +0 -0
  23. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/formats/errors.py +0 -0
  24. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/formats/formats.py +0 -0
  25. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/formats/utils.py +0 -0
  26. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/inference/__init__.py +0 -0
  27. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/inference/patchifier.py +0 -0
  28. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/inference/predictor.py +0 -0
  29. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/metrics/__init__.py +0 -0
  30. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/metrics/available_metrics.py +0 -0
  31. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/metrics/base_metric.py +0 -0
  32. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/metrics/func.py +0 -0
  33. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/metrics/matcher.py +0 -0
  34. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/metrics/utils.py +0 -0
  35. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/__init__.py +0 -0
  36. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/basemodel.py +0 -0
  37. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/mask2former/__init__.py +0 -0
  38. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/mask2former/mask2former.py +0 -0
  39. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/smp/__init__.py +0 -0
  40. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/smp/errors.py +0 -0
  41. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/smp/smp.py +0 -0
  42. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/yolov8/__init__.py +0 -0
  43. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/yolov8/errors.py +0 -0
  44. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/yolov8/utils.py +0 -0
  45. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/yolov8/yolov8.py +0 -0
  46. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/yolov8seg/__init__.py +0 -0
  47. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/yolov8seg/errors.py +0 -0
  48. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/yolov8seg/utils.py +0 -0
  49. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/models/yolov8seg/yolov8seg.py +0 -0
  50. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/preprocessing/__init__.py +0 -0
  51. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/preprocessing/image.py +0 -0
  52. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/preprocessing/preprocessing.py +0 -0
  53. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/train/__init__.py +0 -0
  54. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/train/trainer.py +0 -0
  55. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/utils/__init__.py +0 -0
  56. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/utils/errors.py +0 -0
  57. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools/utils/visualization.py +0 -0
  58. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools.egg-info/SOURCES.txt +0 -0
  59. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools.egg-info/dependency_links.txt +0 -0
  60. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools.egg-info/requires.txt +0 -0
  61. {deepvisiontools-0.0.2 → deepvisiontools-0.0.3}/src/deepvisiontools.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: deepvisiontools
3
- Version: 0.0.2
3
+ Version: 0.0.3
4
4
  Summary: Overlay of PyTorch to generalize trainning and inference processes for detection & instance segmentation tasks.
5
5
  Author-email: Lucas Bernigaud Samatan <lucas.bernigaud-samatan@inrae.fr>, Jordan Bernigaud Samatan <jordan.bernigaud-samatan@inrae.fr>
6
6
  Maintainer-email: Jordan Bernigaud Samatan <jordan.bernigaud-samatan@inrae.fr>
@@ -4,7 +4,7 @@ from deepvisiontools.data.dataset import DeepVisionDataset, DeepVisionLoader
4
4
  from deepvisiontools.utils import visualization
5
5
  from deepvisiontools.inference.predictor import Predictor
6
6
 
7
- __version__ = "0.0.2"
7
+ __version__ = "0.0.3"
8
8
 
9
9
  __all__ = (
10
10
  Configuration,
@@ -176,7 +176,10 @@ class SemanticMaskData(BaseData):
176
176
  if torch.count_nonzero(new_val == cls) < m_min_size:
177
177
  new_val[new_val == cls] = 0
178
178
  if self._scores != None:
179
- new_scores[:, new_val[:] == cls] = 0
179
+ if Configuration().num_classes == 1:
180
+ new_scores[new_val == cls] = 0
181
+ else:
182
+ new_scores[:, new_val[:] == cls] = 0
180
183
  else:
181
184
  # class is preserved and written in present tensor
182
185
  present.append(i - 1)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: deepvisiontools
3
- Version: 0.0.2
3
+ Version: 0.0.3
4
4
  Summary: Overlay of PyTorch to generalize trainning and inference processes for detection & instance segmentation tasks.
5
5
  Author-email: Lucas Bernigaud Samatan <lucas.bernigaud-samatan@inrae.fr>, Jordan Bernigaud Samatan <jordan.bernigaud-samatan@inrae.fr>
6
6
  Maintainer-email: Jordan Bernigaud Samatan <jordan.bernigaud-samatan@inrae.fr>