Исправление бага с размерностью при обучении

2 days ago · 3bf93aab3f
parent 7ad17bb4c4
commit 3bf93aab3f
1 changed files with 1 additions and 1 deletions
--- a/train_scripts/create_dataset_overlay.py
+++ b/train_scripts/create_dataset_overlay.py
@ -107,7 +107,7 @@ def to_uint8(channel):


 def save_notebook_style_png(path, channel):
-    fig = plt.figure()
+    fig = plt.figure(figsize=(16, 16))
    plt.imshow(channel)
    plt.savefig(path)
    plt.clf()