From 987c2268490a6220ac14a1325e76774c60ea4f47 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 29 Aug 2020 15:13:58 -0700
Subject: [PATCH] seaborn plotting correlogram addition

---
 requirements.txt |  3 ++-
 utils/general.py | 13 +++++++++++++
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 7e62ccf..50efbd1 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -23,4 +23,5 @@ tqdm>=4.41.0
 # scikit-learn==0.19.2  # for coreml quantization
 
 # extras --------------------------------------
-# thop
+# thop  # FLOPS computation
+# seaborn  # plotting
diff --git a/utils/general.py b/utils/general.py
index 36d52e3..482b2c7 100755
--- a/utils/general.py
+++ b/utils/general.py
@@ -1188,6 +1188,19 @@ def plot_labels(labels, save_dir=''):
     plt.savefig(Path(save_dir) / 'labels.png', dpi=200)
     plt.close()
 
+    # seaborn correlogram
+    try:
+        import seaborn as sns
+        import pandas as pd
+        x = pd.DataFrame(b.transpose(), columns=['x', 'y', 'width', 'height'])
+        sns.pairplot(x, corner=True, diag_kind='hist', kind='scatter', markers='o',
+                     plot_kws=dict(s=3, edgecolor=None, linewidth=1, alpha=0.02),
+                     diag_kws=dict(bins=50))
+        plt.savefig(Path(save_dir) / 'labels_correlogram.png', dpi=200)
+        plt.close()
+    except Exception as e:
+        pass
+
 
 def plot_evolution(yaml_file='data/hyp.finetune.yaml'):  # from utils.general import *; plot_evolution()
     # Plot hyperparameter evolution results in evolve.txt