th0m45s5helby

OpenCV2

Feb 26th, 2021 (edited)
125
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.06 KB | None | 0 0
  1. import cv2
  2. import csv
  3. import numpy as np
  4. def pic_data_strip_csv():
  5.    img = cv2.imread(r"C:\Users\sudher\Desktop\image.jpg")
  6.    img_height, img_width = img.shape[:2]
  7.    for x in range (0, img_height,1):
  8.       for y in range (0,img_width,1):
  9.          pixel_value = img[x,y,:]
  10.          with open(r"C:\Users\sudher\Desktop\pic.csv",mode="a") as pic_data:
  11.             pic_data_writer = csv.writer(pic_data, delimiter=",", quotechar='"', quoting=csv.QUOTE_MINIMAL)
  12.             write_pic_data = pic_data_writer.writerow([x,y,pixel_value])
  13.    return()
  14. def pic_data_strip_dict_csv():
  15.    img = cv2.imread(r"C:\Users\sudher\Desktop\image.jpg")
  16.    img_height, img_width = img.shape[:2]
  17.    with open(r"C:\Users\sudher\Desktop\pic_dict.csv",mode="a",newline="") as pic_data:
  18.       field_names = ["x","y","bgr"]
  19.       pic_data_writer = csv.DictWriter(pic_data, fieldnames=field_names)
  20.       pic_data_writer.writeheader()
  21.       for x in range (0, img_height,1):
  22.          for y in range (0,img_width,1):
  23.             pixel_value = img[x,y,:]
  24.             write_pic_data = pic_data_writer.writerow({"x":x,"y":y,"bgr":pixel_value})
  25.    return()
  26. '''Sudheer'''
  27. def create_pic_from_csv():
  28.    image = np.zeros((512,512,3), np.uint8)
  29.    with open(r"C:\Users\sudher\Desktop\pic_dict.csv", newline="") as csvfile:
  30.       reader = csv.DictReader(csvfile)
  31.       for row in reader:
  32.          x = row["x"]
  33.          x = int(x)
  34.          y = row["y"]
  35.          y = int(y)
  36.          bgr = row["bgr"]
  37.          #gets the BGR value out of the []
  38.          bgr_edit = str(bgr)
  39.          print(bgr_edit)
  40.          head,sep,tail = bgr_edit.partition(" ")
  41.          b = head.strip("[")
  42.          bgr_tail = tail.strip()
  43.          bgr_tail = bgr_tail.replace("]","")
  44.          head,sep,tail = bgr_tail.partition(" ")
  45.          g = head
  46.          r = tail
  47.          b = int(b)
  48.          g = int(g)
  49.          r = int(r)
  50.          image[x,y]=[b,g,r]
  51.       cv2.imshow("xkcd 1696",image)
  52.       cv2.waitKey(0)
  53.       cv2.destroyAllWindows()
  54.    return()
  55. pic_data_strip_csv()
  56. pic_data_strip_dict_csv()
  57. create_pic_from_csv()
  58.  
Advertisement
Add Comment
Please, Sign In to add comment