.

main.py

@@ -75,16 +75,17 @@ async def root():
 
 @app.post("/segmentation")
 async def segmentation(file: UploadFile = File(...)):
-    global image_dataset
-    a = image_dataset
+    image_dataset = []
+    for file in os.listdir("./images"):
+        image_dataset.append(cv2.resize(cv2.imread('./images/' + file), (160, 544)))
     contents = await file.read()
     image = Image.open(BytesIO(contents))
     open_cv_image = np.array(image)
     open_cv_image = cv2.resize(open_cv_image, (160, 544))
     print(type(image_dataset))
-    a.insert(0, open_cv_image)
-    a = np.transpose(image_dataset, (0, 3, 1, 2))
-    dataset = CustomDataset(a)
+    image_dataset.insert(0, open_cv_image)
+    image_dataset = np.transpose(image_dataset, (0, 3, 1, 2))
+    dataset = CustomDataset(image_dataset)
     dataloader = DataLoader(dataset, batch_size=16, shuffle=False, num_workers=0)
 
     try: