fix some issues of #173

Author: phborba

pytorch_segmentation_models_trainer/dataset_loader/dataset.py

@@ -416,7 +416,7 @@ def load_masks(self, idx):
             }
         if self.return_crossfield_mask:
             mask_dict[self.crossfield_mask_key] = self.load_image(
-                idx, key=self.crossfield_mask_key, is_mask=True
+                idx, key=self.crossfield_mask_key, is_mask=False
             )
         if self.return_distance_mask:
             mask_dict[self.distance_mask_key] = self.load_image(

pytorch_segmentation_models_trainer/utils/polygon_utils.py

@@ -113,7 +113,7 @@ def build_crossfield(polygons, shape, transform, line_width=2):
     for polygon in polygons:
         # --- edges:
         edge_vect_array = np.diff(polygon, axis=0)
-        edge_angle_array = np.angle(edge_vect_array[:, 0] + 1j * edge_vect_array[:, 1])
+        edge_angle_array = np.angle(edge_vect_array[:, 1] + 1j * edge_vect_array[:, 0])
         neg_indices = np.where(edge_angle_array < 0)
         edge_angle_array[neg_indices] += np.pi