Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -34,19 +34,7 @@ if not torch.cuda.is_available():
|
|
34 |
|
35 |
predictor = DefaultPredictor(cfg)
|
36 |
my_metadata = MetadataCatalog.get("car_part_merged_dataset_val")
|
37 |
-
my_metadata.thing_classes =
|
38 |
-
'back_glass',
|
39 |
-
'back__door',
|
40 |
-
'back__light',
|
41 |
-
'front_bumper',
|
42 |
-
'front_glass',
|
43 |
-
'front__door',
|
44 |
-
'front__light',
|
45 |
-
'hood',
|
46 |
-
'_mirror',
|
47 |
-
'tailgate',
|
48 |
-
'trunk',
|
49 |
-
'wheel']
|
50 |
|
51 |
def inference(image):
|
52 |
print(image.height)
|
|
|
34 |
|
35 |
predictor = DefaultPredictor(cfg)
|
36 |
my_metadata = MetadataCatalog.get("car_part_merged_dataset_val")
|
37 |
+
my_metadata.thing_classes =['back_bumper','back_glass','back__door','back__light','front_bumper','front_glass','front__door','front__light','hood','_mirror','tailgate','trunk','wheel']
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
38 |
|
39 |
def inference(image):
|
40 |
print(image.height)
|