Fix sample visualization, remove existing stats for rangeland_weed_au…

…stralia
Project-AgML · May 1, 2024 · 2524c1e · 2524c1e
1 parent fe6a3ee
commit 2524c1e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/agml/_assets/public_datasources.json b/agml/_assets/public_datasources.json
@@ -254,18 +254,6 @@
         "annotation_format": "directory_names",
         "n_images": "17509",
         "docs_url": "https://github.com/AlexOlsen/DeepWeeds",
-        "stats": {
-            "mean": [
-                0.3785816431045532,
-                0.38957422971725464,
-                0.3797682821750641
-            ],
-            "std": [
-                0.22421954572200775,
-                0.22450360655784607,
-                0.22274591028690338
-            ]
-        },
         "classes": {
             "0": "Chinee Apple",
             "1": "Lantana",

diff --git a/agml/viz/labels.py b/agml/viz/labels.py
@@ -116,7 +116,11 @@ def show_images_and_labels(images,
         plt.setp(ax.spines.values(), visible = False)
         ax.set_xlabel(label)
 
+        # decrease the label size (if it's too big)
+        ax.xaxis.label.set_size(8)
+
     # Display and return the image.
+    fig.tight_layout()
     image = convert_figure_to_image()
     if not kwargs.get('no_show', False):
         _ = display_image(image)