Merge pull request #54 from NUTFes/feat/yamada/53-create-number-category

学習用ナンバープレート作成

raspberryPi/.gitignore

@@ -1 +1 @@
-client_secrets.json
+*.json

raspberryPi/create_number_plate.py

@@ -1,6 +1,11 @@
 from number_plate import NumberPlate
+from dotenv import load_dotenv
+import os
 import cv2
 
+load_dotenv()
+HOME_DIR = os.environ['HOME_DIR']
+
 category=0 # 0:普通車（自家用） 1:普通車（事業用） 2:軽自動車（自家用） 3:軽自動車（事業用）
 
 
@@ -10,5 +15,11 @@
 
 # imgに画像のデータがはいっているから、これを保存すると画像になる
 # 実行するとtest.pngが生成されるからやってみて
-cv2.imwrite('./test.jpg', img)
+#cv2.imwrite('./test.jpg', img)
 
+for i in range (3):
+    category=i
+    for j in range (10000):
+        img = number_plate.generate(category)
+        # print(HOME_DIR +'/raspberryPi/resorces/create_nuber_plates/img{i}.{j}.jpg')
+        cv2.imwrite(HOME_DIR +'/raspberryPi/resorces/create_nuber_plates/img' + str(i) + "_" + str(j) + '.jpg' ,img)

raspberryPi/image_upload.py

@@ -39,7 +39,7 @@ def check_files(self, folder_name,):
     def upload(self, 
                local_file_path: str,
                save_folder_name: str = 'sample',
-               is_convert : bool=True,
+               is_convert : bool=False,
         ):
 
         if save_folder_name:
@@ -62,9 +62,13 @@ def upload(self,
 
 if __name__ == "__main__":
     g = GoogleDriveFacade()
-    g.upload(
-        local_file_path='hoge.txt',
-        save_folder_name="test",
-        is_convert=True,
+    for i in range (3):
+        for j in range (10000):
+            local_file_path='./resorces/create_nuber_plates/img' + str(i) + "_" + str(j) + '.jpg'
+            print(local_file_path)
+            g.upload(
+                local_file_path='./resorces/create_nuber_plates/img' + str(i) + "_" + str(j) + '.jpg',
+                save_folder_name="test5",
+                is_convert=False,
     )
 

raspberryPi/resorces/create_nuber_plates/.gitignore

@@ -1 +1,2 @@
-*.jpg
+*
+!.gitignore