add dataset

@@ -58,4 +58,10 @@ googel chrome의 확장프로그램을 활용한 구글 이미지 검색 결과 크롤링 프로그램 구현
 라즈베리파이에서 구동되는 코드 추가
 make_img.py에서는 라즈베리파이에 연결된 카메라가 촬영을 하면 그 사진을 32*32*3 사이즈로 resize해준 후 input.txt파일로 변환해줌
 main.cpp에서 convnet 코드 실행
-predict.sh 쉘파일로 전체 실행
\ No newline at end of file
+predict.sh 쉘파일로 전체 실행
+
+
+13. dataset 추가
+image_process.py : cropping을 통해 정사각형 형태를 만들고 32*32 크기로 resize, 사진 반전, 회전을 통해 사진 1개당 8개의 데이터를 생성
+크롤링을 통해 휴지, 소주, 캔음료, 라면, 삼각김밥 각각 500개씩의 이미지를 다운받음
+한 카테고리당 4000개의 데이터를 생성
\ No newline at end of file

+import os, sys
+from PIL import Image
+
+def make_img(filename):
+    try:
+        with Image.open(filename + '.png') as im:
+            print("File ("+filename +".png)")
+
+            if im.width<im.height :
+                crop_size = (im.height-im.width)/2
+                area = (0, int(crop_size), im.width, int(crop_size)+im.width)
+
+            elif im.width>im.height :
+                crop_size = (im.width-im.height) / 2
+                area = (int(crop_size), 0, int(crop_size)+im.height, im.height)
+
+            else :
+                area = (0, 0, im.width , im.height)
+
+            cropped_img = im.crop(area)
+            cropped_img.thumbnail((32, 32))
+            cropped_img.save(filename + "_0.png")
+
+            im2 = cropped_img.rotate(90)
+            im2.save(filename + "_90.png")
+
+            im3 = cropped_img.rotate(180)
+            im3.save(filename + "_180.png")
+
+            im4 = cropped_img.rotate(270)
+            im4.save(filename + "_270.png")
+
+            im_transpose = cropped_img.transpose(Image.FLIP_LEFT_RIGHT)
+            im5 = im_transpose
+            im5.save(filename + "_transpose_0.png")
+
+            im6 = im_transpose.rotate(90)
+            im6.save(filename + "_transpose_90.png")
+
+            im7 = im_transpose.rotate(180)
+            im7.save(filename + "_transpose_180.png")
+
+            im8 = im_transpose.rotate(270)
+            im8.save(filename + "_transpose_270.png")
+
+            os.remove(filename+'.png')
+
+    except IOError:
+        print("No file ("+filename +".png)")
+
+
+
+for i in range(0,1000):
+        filename =  str(i)
+        make_img(filename)
+        filename = '0' + str(i)
+        make_img(filename)