minor bugfixes (#1966)

Co-authored-by: Ofri Masad <ofrimasad@users.noreply.github.com> Co-authored-by: Eugene Khvedchenya <ekhvedchenya@gmail.com>
Deci-AI · May 16, 2024 · dd56e79 · dd56e79
1 parent f8cc94a
commit dd56e79
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/super_gradients/training/datasets/detection_datasets/detection_dataset.py b/src/super_gradients/training/datasets/detection_datasets/detection_dataset.py
@@ -493,7 +493,7 @@ def plot(
 
             # Plot `max_samples_per_plot` images.
             for img_i in range(max_samples_per_plot):
-                index = img_i + plot_i * 16
+                index = img_i + plot_i * max_samples_per_plot
 
                 # LOAD IMAGE/TARGETS
                 if plot_transformed_data:
@@ -518,7 +518,7 @@ def plot(
 
                 image = DetectionVisualization.visualize_image(image_np=image, class_names=self.classes, target_boxes=targets_label_xyxy, gt_alpha=1)
 
-                plt.subplot(n_subplot, n_subplot, img_i + 1).imshow(image[:, :, ::-1])
+                plt.subplot(n_subplot, n_subplot, img_i + 1).imshow(image)
                 plt.imshow(image)
                 plt.axis("off")