init

VitTrainer.py

@@ -1,4 +1,5 @@
 # tensor - numpy - PILImage 변환 (https://qlsenddl-lab.tistory.com/37)
+from PIL import Image
 
 import os
 os.environ['KMP_DUPLICATE_LIB_OK']='True'
@@ -214,6 +215,8 @@ class VitTrainer:
 
 if __name__ == "__main__":
 
+    image = Image.open("img.png")
+
     PROJECT_HOME = os.getcwd()
     RESOURCE_PATH = os.path.join(PROJECT_HOME, "resources")
     model_path = os.path.join(RESOURCE_PATH, "model")