Update README.md
Browse files
README.md
CHANGED
|
@@ -86,12 +86,15 @@ def colorize_image(img_path, model):
|
|
| 86 |
return np.stack(rgb_imgs, axis=0)
|
| 87 |
|
| 88 |
#Example image path
|
|
|
|
| 89 |
img_path = "/path/to/your/image.jpg" # Replace with your image path
|
| 90 |
|
| 91 |
#Perform inference
|
|
|
|
| 92 |
colorized_images = colorize_image(img_path, G_net)
|
| 93 |
|
| 94 |
#Display the result
|
|
|
|
| 95 |
plt.imshow(colorized_images[0])
|
| 96 |
plt.axis("off")
|
| 97 |
plt.show()
|
|
|
|
| 86 |
return np.stack(rgb_imgs, axis=0)
|
| 87 |
|
| 88 |
#Example image path
|
| 89 |
+
|
| 90 |
img_path = "/path/to/your/image.jpg" # Replace with your image path
|
| 91 |
|
| 92 |
#Perform inference
|
| 93 |
+
|
| 94 |
colorized_images = colorize_image(img_path, G_net)
|
| 95 |
|
| 96 |
#Display the result
|
| 97 |
+
|
| 98 |
plt.imshow(colorized_images[0])
|
| 99 |
plt.axis("off")
|
| 100 |
plt.show()
|