(img_path)
| 6 | import numpy as np |
| 7 | |
| 8 | def convert_color(img_path): |
| 9 | plt.figure() |
| 10 | img = cv2.imread(img_path, cv2.IMREAD_GRAYSCALE) |
| 11 | plt.imsave(img_path, img, cmap=plt.get_cmap('viridis')) |
| 12 | plt.close() |
| 13 | |
| 14 | |
| 15 | def save_tensor(tensor, path, pad_value=254.0,): |