zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

seaborn plotting correlogram addition

This commit is contained in:
Glenn Jocher 2020-08-29 15:13:58 -07:00
parent 08e97a2f88
commit 987c226849
2 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
requirements.txt
View File

@ -23,4 +23,5 @@ tqdm>=4.41.0
# scikit-learn==0.19.2 # for coreml quantization # scikit-learn==0.19.2 # for coreml quantization
# extras -------------------------------------- # extras --------------------------------------
# thop # thop # FLOPS computation
# seaborn # plotting

13
utils/general.py
View File

@ -1188,6 +1188,19 @@ def plot_labels(labels, save_dir=''):
plt.savefig(Path(save_dir) / 'labels.png', dpi=200) plt.savefig(Path(save_dir) / 'labels.png', dpi=200)
plt.close() 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() def plot_evolution(yaml_file='data/hyp.finetune.yaml'): # from utils.general import *; plot_evolution()
# Plot hyperparameter evolution results in evolve.txt # Plot hyperparameter evolution results in evolve.txt