step1

Author: yanx27

make_source.py

@@ -95,7 +95,7 @@
            int(head_cord[0] - crop_size): int(head_cord[0] + crop_size), :]
     plt.imshow(head)
     plt.savefig(str(test_head_dir.joinpath('pose_{}.jpg'.format(idx))))
-
+    plt.clf()
     cv2.imwrite(str(test_img_dir.joinpath('{:05}.png'.format(idx))), img)
     cv2.imwrite(str(test_label_dir.joinpath('{:05}.png'.format(idx))), label)
     if idx % 100 == 0 and idx != 0: