| milind-u | 7d834eb | 2021-11-20 08:46:11 -0800 | [diff] [blame] | 1 | #!/usr/bin/python3 |
| 2 | |
| 3 | from absl import app, flags |
| 4 | import cv2 as cv |
| 5 | import glog |
| 6 | import matplotlib.pyplot as plt |
| 7 | |
| 8 | flags.DEFINE_bool("hsv", False, "Displays the image in hsv") |
| 9 | |
| 10 | |
| 11 | def main(argv): |
| 12 | glog.check_eq(len(argv), 2, "Expected one image filename as an argument") |
| 13 | bgr_img = cv.imread(argv[1]) |
| 14 | img = cv.cvtColor( |
| 15 | bgr_img, cv.COLOR_BGR2HSV if flags.FLAGS.hsv else cv.COLOR_BGR2RGB) |
| 16 | plt.imshow(img) |
| 17 | plt.show() |
| 18 | |
| 19 | |
| 20 | if __name__ == "__main__": |
| 21 | app.run(main) |