code-loader 0.2.66__tar.gz → 0.2.67__tar.gz

{code_loader-0.2.66 → code_loader-0.2.67}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 0.2.66
+Version: 0.2.67
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT

{code_loader-0.2.66 → code_loader-0.2.67}/code_loader/helpers/detection/yolo/loss.py

@@ -165,13 +165,13 @@ class YoloLoss:
                                 gj: List[torch.Tensor], gi: List[torch.Tensor],
                                 target: List[torch.Tensor]) -> Tuple[tf.Tensor, tf.Tensor]:
         # temp
+        assert self.anchors is not None
         gt_class = tf.ones((batch_size,
                             self.anchors.shape[1], *self.feature_maps[i]), dtype=tf.int32) * self.background_label
         if len(b[i]) > 0:
             gt_class = tf.tensor_scatter_nd_update(gt_class, torch.stack([b[i], a[i], gj[i], gi[i]]).T.numpy(),
                                                    target[i][:, 1])
         conf_t_tensor = tf.reshape(gt_class, [gt_class.shape[0], -1])
-        assert self.anchors is not None
         gt_loc = tf.zeros((batch_size, len(self.anchors[i]), *self.feature_maps[i], 4), dtype=tf.float32)
         if len(b[i]) > 0:
             gt_loc = tf.tensor_scatter_nd_update(gt_loc, torch.stack([b[i], a[i], gj[i], gi[i]]).T.numpy(),
@@ -181,6 +181,7 @@ class YoloLoss:
 
     def get_yolo_match(self, batch_size: int, y_true: tf.Tensor, loc_data: List[tf.Tensor], conf_data: List[tf.Tensor]) \
             -> Tuple[List[torch.Tensor], ...]:
+        assert self.anchors is not None
         yolo_targets: List[NDArray[np.float32]] = []
         scales_num = len(loc_data)
         for i in range(batch_size):

{code_loader-0.2.66 → code_loader-0.2.67}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "0.2.66"
+version = "0.2.67"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"