Spaces:

sir3mat
/

SchoolInAiProjectWork

Build error

Matteo Sirri commited on Nov 21, 2022

Commit

dea24b0

1 Parent(s): 7a129e5

chore: change input folders

Files changed (8) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import logging
 import torchvision
 from torchvision.models.detection.faster_rcnn import fasterrcnn_resnet50_fpn_v2
 from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
-from configs.path_cfg import  OUTPUT_DIR
 from src.detection.graph_utils import add_bbox
 from src.detection.vision import presets
 logging.getLogger('PIL').setLevel(logging.CRITICAL)
@@ -48,7 +48,7 @@ def detect_with_resnet50Model_baseline(image):
 title = "Performance comparision of Faster R-CNN for people detection with syntetic data"
 description = "<p style='text-align: center'>Performance comparision of Faster RCNN models for people detection using MOTSynth and MOT17"
-examples = ["example1.jpg", "example2.jpg"]
 io_baseline = gr.Interface(detect_with_resnet50Model_baseline, gr.Image(type="pil"), gr.Image(
     type="file", shape=(1920, 1080), label="Baseline Faster RCNN Model pretrained on COCO dataset"))

 import torchvision
 from torchvision.models.detection.faster_rcnn import fasterrcnn_resnet50_fpn_v2
 from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
+from configs.path_cfg import OUTPUT_DIR
 from src.detection.graph_utils import add_bbox
 from src.detection.vision import presets
 logging.getLogger('PIL').setLevel(logging.CRITICAL)
 title = "Performance comparision of Faster R-CNN for people detection with syntetic data"
 description = "<p style='text-align: center'>Performance comparision of Faster RCNN models for people detection using MOTSynth and MOT17"
+examples = "/input_examples"
 io_baseline = gr.Interface(detect_with_resnet50Model_baseline, gr.Image(type="pil"), gr.Image(
     type="file", shape=(1920, 1080), label="Baseline Faster RCNN Model pretrained on COCO dataset"))