Update app.py

app.py

@@ -73,11 +73,18 @@ def classifyCar(im):
     label = "fail"
   try:
     outputs = predictor(im)
-    v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
+    v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1)
     out = v.draw_instance_predictions(outputs["instances"])
+    
     car_class_true = outputs["instances"].pred_classes == 2
-    boxes = outputs["instances"].pred_boxes[car_class_true]
-    im2 = Image.fromarray(np.uint8(out.get_image())).convert('RGB').crop(list(outputs["instances"].pred_boxes)[1].numpy())
+    boxes = list(outputs["instances"].pred_boxes[car_class_true])
+    max_idx = torch.tensor([(x[2] - x[0])*(x[3] - x[1]) for x in boxes]).argmax().item()
+    
+    im2 = Image.fromarray(np.uint8(out.get_image())).convert('RGB').crop(boxes[max_idx].to(torch.int64).numpy())
+    
+    carTransforms = transforms.Compose([transforms.Resize((224, 224))])
+    im2 = carTransforms(im2)
+
     label = "success"
   except:
     label = "fail2"