Task5-Prediction

Suddhasil/check.py

@@ -0,0 +1,35 @@
+import cv2
+img = cv2.imread('SampleImage.jpg')
+img2= cv2.resize(img, (300,250))
+cv2.imshow("Original Image",img2)
+imCopy=img2.copy()
+gray=cv2.cvtColor(img2 ,cv2.COLOR_BGR2GRAY)
+blur=cv2.GaussianBlur(gray,(5,5),0)
+cv2.imshow("Blur Image",blur)
+
+ret, thresh1= cv2.threshold(blur,10,225,cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
+cv2.imshow("Thresh_bin_inv", thresh1)
+
+ret, thresh2= cv2.threshold(blur,10,225,cv2.THRESH_TOZERO + cv2.THRESH_OTSU)
+cv2.imshow("Thresh_tozero", thresh2)
+
+ret, thresh3= cv2.threshold(blur,10,225,cv2.THRESH_TOZERO_INV + cv2.THRESH_OTSU)
+cv2.imshow("Thresh_tozero_inv", thresh3)
+
+ret, thresh4= cv2.threshold(blur,10,225,cv2.THRESH_TRUNC + cv2.THRESH_OTSU)
+cv2.imshow("Thresh_trunc", thresh4)
+
+ret, thresh5= cv2.threshold(blur,10,225,cv2.THRESH_BINARY + cv2.THRESH_OTSU)
+cv2.imshow("Thresh_bin", thresh5)
+
+contours, hierarchy= cv2.findContours(thresh1, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
+cv2.drawContours(imCopy, contours, -1, (0,255,0))
+cv2.imshow('Tree Contour',imCopy)
+
+contours, hierarchy= cv2.findContours(thresh1, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)
+cv2.drawContours(imCopy, contours, -1, (0,255,0))
+cv2.imshow('List Contour',imCopy)
+
+interrupt=cv2.waitKey(0) & 0xFF
+if interrupt==27:
+    cv2.destroyAllWindows()

Suddhasil/collect-data.py

@@ -0,0 +1,79 @@
+import cv2
+import os
+
+if not os.path.exists("data"):
+    os.makedirs("data")
+    os.makedirs("data/train")
+    os.makedirs("data/test")
+    os.makedirs("data/train/0")
+    os.makedirs("data/train/1")
+    os.makedirs("data/train/2")
+    os.makedirs("data/train/3")
+    os.makedirs("data/train/4")
+    os.makedirs("data/train/5")
+    os.makedirs("data/test/0")
+    os.makedirs("data/test/1")
+    os.makedirs("data/test/2")
+    os.makedirs("data/test/3")
+    os.makedirs("data/test/4")
+    os.makedirs("data/test/5")
+
+mode = 'train'
+directory = 'data/'+mode+'/' 
+
+cap=cv2.VideoCapture(0)
+
+while True:
+    _, frame = cap.read()
+    frame = cv2.flip(frame, 1)
+
+    cv2.putText(frame, "SuddhasilC/Expressando", (175, 450), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (225,255,255), 3)
+
+    count = {'zero': len(os.listdir(directory+"/0")),
+            'one': len(os.listdir(directory+"/1")),
+            'two': len(os.listdir(directory+"/2")),
+            'three': len(os.listdir(directory+"/3")),
+            'four': len(os.listdir(directory+"/4")),
+            'five': len(os.listdir(directory+"/5"))}
+
+    cv2.putText(frame, "MODE : "+mode, (30, 50), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (225,255,255), 1)
+    cv2.putText(frame, "IMAGE COUNT", (10, 100), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (225,255,255), 1)
+    cv2.putText(frame, "ZERO : "+str(count['zero']), (10, 120), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (255,255,255), 1)
+    cv2.putText(frame, "ONE : "+str(count['one']), (10, 140), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (255,255,255), 1)
+    cv2.putText(frame, "TWO : "+str(count['two']), (10, 160), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (255,255,255), 1)
+    cv2.putText(frame, "THREE : "+str(count['three']), (10, 180), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (255,255,255), 1)
+    cv2.putText(frame, "FOUR : "+str(count['four']), (10, 200), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (255,255,255), 1)
+    cv2.putText(frame, "FIVE : "+str(count['five']), (10, 220), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (255,255,255), 1)
+
+    x1 = int(0.5*frame.shape[1])
+    y1 = 10
+    x2 = frame.shape[1]-10
+    y2 = int(0.5*frame.shape[1])
+    cv2.rectangle(frame, (x1-1, y1-1), (x2+1, y2+1), (255,0,0) ,3)
+    roi = frame[y1:y2, x1:x2]
+    roi = cv2.resize(roi, (200, 200))
+    cv2.putText(frame, "R.O.I", (440, 350), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (0,225,0), 3)
+    cv2.imshow("Frame", frame)
+
+    roi = cv2.cvtColor(roi, cv2.COLOR_BGR2GRAY)
+    _, roi = cv2.threshold(roi, 120, 255, cv2.THRESH_BINARY)
+    cv2.imshow("ROI", roi)
+
+    interrupt = cv2.waitKey(10)
+    if interrupt & 0xFF == 27:
+        break
+    if interrupt & 0xFF == ord('0'):
+        cv2.imwrite(directory+'0/'+str(count['zero'])+'.jpg', roi)
+    if interrupt & 0xFF == ord('1'):
+        cv2.imwrite(directory+'1/'+str(count['one'])+'.jpg', roi)
+    if interrupt & 0xFF == ord('2'):
+        cv2.imwrite(directory+'2/'+str(count['two'])+'.jpg', roi)
+    if interrupt & 0xFF == ord('3'):
+        cv2.imwrite(directory+'3/'+str(count['three'])+'.jpg', roi)
+    if interrupt & 0xFF == ord('4'):
+        cv2.imwrite(directory+'4/'+str(count['four'])+'.jpg', roi)
+    if interrupt & 0xFF == ord('5'):
+        cv2.imwrite(directory+'5/'+str(count['five'])+'.jpg', roi)
+
+cap.release()
+cv2.destroyAllWindows()