Fix visualization title bug (#9500)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-19 20:38:11 +02:00 · 2022-09-19 20:38:11 +02:00 · 0171198f38
parent 0b724c5b85
commit 0171198f38
1 changed files with 0 additions and 1 deletions
--- a/utils/plots.py
+++ b/utils/plots.py
@ -204,7 +204,6 @@ def feature_visualization(x, module_type, stage, n=32, save_dir=Path('runs/detec
                ax[i].axis('off')

            LOGGER.info(f'Saving {f}... ({n}/{channels})')
-            plt.title('Features')
            plt.savefig(f, dpi=300, bbox_inches='tight')
            plt.close()
            np.save(str(f.with_suffix('.npy')), x[0].cpu().numpy())  # npy save