Inital Commit

work.py

@@ -0,0 +1,69 @@
+'''
+Prototype Scrolling Ticker Algorithm in Python
+For Google Summer of Code 2018 (Red Hen Labs)
+Burhan Ul Tayyab
+'''
+
+import cv2
+import numpy as np 
+import sys
+
+cap = cv2.VideoCapture(sys.argv[1])
+thresh = 70
+kernel = np.ones((8,5),np.uint8)
+name = "img"
+k = 0
+t = 0
+
+
+while(cap.isOpened()):
+    #Reading Frames
+    ret, frame = cap.read()
+    #Creating a Rectangle (Bounding Box) over Scrolling Tickers
+    cv2.rectangle(frame, (100, 525), (715, 550), (255,0,0), 2)
+    #cv2.imshow('frame',frame)
+    #Cropping the image
+    crop_img = frame[525:525+25, 100:100+615]
+    cv2.imshow('framex',crop_img)
+
+    #Detecting Blue Areas for Ticker Stopping Point
+    hsv = cv2.cvtColor(crop_img, cv2.COLOR_BGR2HSV)
+    lower_red = np.array([110,50,50])
+    upper_red = np.array([130,255,255])
+    mask = cv2.inRange(hsv, lower_red, upper_red)
+    res = cv2.bitwise_and(crop_img,crop_img, mask= mask)
+
+    #Graying and Thresholding for finding Contours
+    gray=cv2.cvtColor(res,cv2.COLOR_BGR2GRAY)
+    ret,thresh1 = cv2.threshold(gray,thresh,255,cv2.THRESH_BINARY)
+    closing = cv2.morphologyEx(thresh1, cv2.MORPH_CLOSE, kernel)
+
+    t = t + 1
+    #t,contours, hier = cv2.findContours(thresh1,cv2.RETR_LIST,cv2.CHAIN_APPROX_SIMPLE)
+    m, contours, _ = cv2.findContours(closing, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)
+    for cnt in contours:
+        k = k + 1
+        rect = cv2.boundingRect(cnt)
+        if rect[0] > 300 and rect[0] < 450:
+          print(rect[0])
+          print(cnt[0])
+          cv2.drawContours(crop_img,[cnt],0,(0,255,0),2)
+          record = 1
+          cv2.imshow('framxds', crop_img)
+        elif rect[0] < 150:
+          #vis = np.concatenate((crop_img, crop_img), axis=1)
+          save = name + str(k) + str(t) + ".jpg"
+          cv2.imwrite(save, crop_img)
+
+
+          cv2.imshow('framxdss', vis)
+    cv2.imshow('res',closing)
+    cv2.imshow('fx',frame)
+
+
+
+    if cv2.waitKey(1) & 0xFF == ord('q'):
+        break
+
+cap.release()
+cv2.destroyAllWindows()