PytorchWildlife 1.2.1__tar.gz → 1.2.2__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 (51) hide show
  1. {pytorchwildlife-1.2.1/PytorchWildlife.egg-info → pytorchwildlife-1.2.2}/PKG-INFO +1 -1
  2. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/ultralytics_based/yolov8_base.py +8 -1
  3. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2/PytorchWildlife.egg-info}/PKG-INFO +1 -1
  4. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/setup.py +1 -1
  5. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/LICENSE +0 -0
  6. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PW_FT_classification/__init__.py +0 -0
  7. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PW_FT_classification/main.py +0 -0
  8. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/__init__.py +0 -0
  9. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/data/__init__.py +0 -0
  10. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/data/datasets.py +0 -0
  11. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/data/transforms.py +0 -0
  12. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/__init__.py +0 -0
  13. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/__init__.py +0 -0
  14. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/base_classifier.py +0 -0
  15. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/resnet_base/__init__.py +0 -0
  16. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/resnet_base/amazon.py +0 -0
  17. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/resnet_base/base_classifier.py +0 -0
  18. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/resnet_base/custom_weights.py +0 -0
  19. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/resnet_base/opossum.py +0 -0
  20. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/resnet_base/serengeti.py +0 -0
  21. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/timm_base/DFNE.py +0 -0
  22. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/timm_base/Deepfaune.py +0 -0
  23. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/timm_base/__init__.py +0 -0
  24. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/classification/timm_base/base_classifier.py +0 -0
  25. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/__init__.py +0 -0
  26. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/base_detector.py +0 -0
  27. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/herdnet/__init__.py +0 -0
  28. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/herdnet/animaloc/__init__.py +0 -0
  29. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/herdnet/animaloc/data/__init__.py +0 -0
  30. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/herdnet/animaloc/data/patches.py +0 -0
  31. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/herdnet/animaloc/data/types.py +0 -0
  32. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/herdnet/animaloc/eval/__init__.py +0 -0
  33. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/herdnet/animaloc/eval/lmds.py +0 -0
  34. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/herdnet/animaloc/eval/stitchers.py +0 -0
  35. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/herdnet/dla.py +0 -0
  36. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/herdnet/herdnet.py +0 -0
  37. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/herdnet/model.py +0 -0
  38. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/ultralytics_based/Deepfaune.py +0 -0
  39. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/ultralytics_based/__init__.py +0 -0
  40. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/ultralytics_based/megadetectorv5.py +0 -0
  41. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/ultralytics_based/megadetectorv6.py +0 -0
  42. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/models/detection/ultralytics_based/yolov5_base.py +0 -0
  43. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/utils/__init__.py +0 -0
  44. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/utils/misc.py +0 -0
  45. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife/utils/post_process.py +0 -0
  46. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife.egg-info/SOURCES.txt +0 -0
  47. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife.egg-info/dependency_links.txt +0 -0
  48. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife.egg-info/requires.txt +0 -0
  49. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/PytorchWildlife.egg-info/top_level.txt +0 -0
  50. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/README.md +0 -0
  51. {pytorchwildlife-1.2.1 → pytorchwildlife-1.2.2}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: PytorchWildlife
3
- Version: 1.2.1
3
+ Version: 1.2.2
4
4
  Summary: a PyTorch Collaborative Deep Learning Framework for Conservation.
5
5
  Home-page: https://github.com/microsoft/CameraTraps/
6
6
  Author: Andres Hernandez, Zhongqi Miao, Daniela Ruiz Lopez, Isai Daniel Chacon Silva
@@ -144,9 +144,16 @@ class YOLOV8Base(BaseDetector):
144
144
 
145
145
  self.predictor.args.batch = 1
146
146
  self.predictor.args.conf = det_conf_thres
147
+
147
148
  det_results = list(self.predictor.stream_inference([img]))
149
+
150
+ res = self.results_generation(det_results[0], img_path, id_strip)
151
+
152
+ normalized_coords = [[x1 / img_size[1], y1 / img_size[0], x2 / img_size[1], y2 / img_size[0]]
153
+ for x1, y1, x2, y2 in res["detections"].xyxy]
154
+ res["normalized_coords"] = normalized_coords
148
155
 
149
- return self.results_generation(det_results[0], img_path, id_strip)
156
+ return res
150
157
 
151
158
  def batch_image_detection(self, data_path, batch_size=16, det_conf_thres=0.2, id_strip=None):
152
159
  """
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: PytorchWildlife
3
- Version: 1.2.1
3
+ Version: 1.2.2
4
4
  Summary: a PyTorch Collaborative Deep Learning Framework for Conservation.
5
5
  Home-page: https://github.com/microsoft/CameraTraps/
6
6
  Author: Andres Hernandez, Zhongqi Miao, Daniela Ruiz Lopez, Isai Daniel Chacon Silva
@@ -4,7 +4,7 @@ with open('README.md', encoding="utf8") as file:
4
4
  long_description = file.read()
5
5
  setup(
6
6
  name='PytorchWildlife',
7
- version='1.2.1',
7
+ version='1.2.2',
8
8
  packages=find_packages(),
9
9
  url='https://github.com/microsoft/CameraTraps/',
10
10
  license='MIT',
File without changes