| 383 | visualize(image, saturated_3) |
| 384 | |
| 385 | def visualize(original, augmented): |
| 386 | plt.figure(figsize = (20, 10)) |
| 387 | plt.subplot(1, 2, 1) |
| 388 | plt.title("Original Picture", fontsize=50, fontweight='bold') |
| 389 | # plt.axis("off") # 关闭坐标轴显示 |
| 390 | #plt.imshow(original) |
| 391 | |
| 392 | plt.subplot(1, 2, 2) |
| 393 | plt.title("saturation 3", fontsize=50, fontweight='bold') |
| 394 | # plt.axis("off") # 关闭坐标轴显示 |
| 395 | #plt.imshow(augmented) |
| 396 | plt.xticks(fontsize = 30) |
| 397 | plt.yticks(fontsize = 30) |
| 398 | plt.tight_layout() |
| 399 | plt.savefig('./saturation3.png') |
| 400 | #plt.show() |
| 401 | |
| 402 | |
| 403 | if __name__ == '__main__': |