Place a system metadata stamp on readme chart (#405)

pytorch · Nov 20, 2024 · fd4c288 · fd4c288
1 parent 789cead
commit fd4c288
Show file tree

Hide file tree

Showing 4 changed files with 423 additions and 409 deletions.
diff --git a/benchmarks/decoders/benchmark_decoders_library.py b/benchmarks/decoders/benchmark_decoders_library.py
@@ -444,11 +444,11 @@ def retrieve_videos(urls_and_dest_paths):
         urllib.request.urlretrieve(url, path)
 
 
-def plot_data(df_data, plot_path):
+def plot_data(json_data, plot_path):
     plt.rcParams["font.size"] = 18
 
     # Creating the DataFrame
-    df = pd.DataFrame(df_data)
+    df = pd.DataFrame(json_data["experiments"])
 
     # Sorting by video, type, and frame_count
     df_sorted = df.sort_values(by=["video", "type", "frame_count"])
@@ -518,6 +518,14 @@ def plot_data(df_data, plot_path):
         for col in range(video_type_combinations[unique_videos[row]], max_combinations):
             fig.delaxes(axes[row, col])
 
+    plt.gcf().text(
+        0.005,
+        0.87,
+        "\n".join([f"{k}: {v}" for k, v in json_data["system_metadata"].items()]),
+        fontsize=11,
+        bbox=dict(facecolor="white"),
+    )
+
     # Adjust layout to avoid overlap
     plt.tight_layout()
 

diff --git a/benchmarks/decoders/benchmark_readme_chart.png b/benchmarks/decoders/benchmark_readme_chart.png