code-loader 0.2.59.dev1__py3-none-any.whl → 0.2.59.dev2__py3-none-any.whl

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.
@@ -185,7 +185,7 @@ class YoloLoss:
185
185
  scales_num = len(loc_data)
186
186
  for i in range(batch_size):
187
187
  batch_gt = y_true[i][y_true[i, ..., -1] != self.background_label]
188
- yolo_targets.append(np.concatenate([np.ones((1, 1)) * i, batch_gt[:, 4:], batch_gt[:, :4]], axis=1))
188
+ yolo_targets.append(np.concatenate([np.ones((batch_gt.shape[0], 1)) * i, batch_gt[:, 4:], batch_gt[:, :4]], axis=1))
189
189
  yolo_targets_cat: NDArray[np.float32] = np.concatenate(yolo_targets, axis=0)
190
190
  orig_pred = [torch.from_numpy(tf.concat([loc_data[i], conf_data[i]], axis=-1).numpy()) for i in
191
191
  range(scales_num)]
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: code-loader
3
- Version: 0.2.59.dev1
3
+ Version: 0.2.59.dev2
4
4
  Summary:
5
5
  Home-page: https://github.com/tensorleap/code-loader
6
6
  License: MIT
@@ -12,7 +12,7 @@ code_loader/helpers/detection/utils.py,sha256=TCoprd1jOFQOu5hgYlKyIMWKJFirKCOivX
12
12
  code_loader/helpers/detection/yolo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
13
13
  code_loader/helpers/detection/yolo/decoder.py,sha256=EsjebDBya53BXtth3vCwLNAlCVwpaTpG0n8oUB2AJOg,6398
14
14
  code_loader/helpers/detection/yolo/grid.py,sha256=xcsHh7P10_PrEavvJm1JAraxteQujqN42RfMP5nYrts,3132
15
- code_loader/helpers/detection/yolo/loss.py,sha256=oLxRqUEO2GLqZbfwc_-Ta8z0CgIEygEkm5BjRREUIKE,11009
15
+ code_loader/helpers/detection/yolo/loss.py,sha256=1B8qKm1VzPsyFY-BsYJPGOw5QAP6aQqu1iEGw8Gym9U,11025
16
16
  code_loader/helpers/detection/yolo/pytorch_utils.py,sha256=D4Rreaehz3R8YMvLRbPIhbxHt-dBieKLVyv527c_XjI,10454
17
17
  code_loader/helpers/detection/yolo/utils.py,sha256=eV9-IZhglRNGud1UmSckIuA49WRKJitDQhPKK7iIaBc,6563
18
18
  code_loader/leap_binder/__init__.py,sha256=Oe7DNj6RmgX-tMruOrI7GYvf3SyAnzOxugF2e117_Z8,93
@@ -23,7 +23,7 @@ code_loader/metrics/default_metrics.py,sha256=xtvTskR7LC5pyKSlFcvlyBpFY9TSXNGVAi
23
23
  code_loader/utils.py,sha256=WUcM97OuxrhfLCRPoH9EbXrxajNpYgX1CTMc3_PXtYU,1736
24
24
  code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
25
25
  code_loader/visualizers/default_visualizers.py,sha256=HqWx2qfTrroGl2n8Fpmr_4X-rk7tE2oGapjO3gzz4WY,2226
26
- code_loader-0.2.59.dev1.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
27
- code_loader-0.2.59.dev1.dist-info/WHEEL,sha256=vxFmldFsRN_Hx10GDvsdv1wroKq8r5Lzvjp6GZ4OO8c,88
28
- code_loader-0.2.59.dev1.dist-info/METADATA,sha256=UZ7csnXAHfQKqSoF2AGqUA9pRu_ExZOJ4QdzCadYHEk,879
29
- code_loader-0.2.59.dev1.dist-info/RECORD,,
26
+ code_loader-0.2.59.dev2.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
27
+ code_loader-0.2.59.dev2.dist-info/WHEEL,sha256=vxFmldFsRN_Hx10GDvsdv1wroKq8r5Lzvjp6GZ4OO8c,88
28
+ code_loader-0.2.59.dev2.dist-info/METADATA,sha256=yQIUqSccVcjxMudRIkDpJBMzQzqi-EYv9fVp5n1UopA,879
29
+ code_loader-0.2.59.dev2.dist-info/RECORD,,