ruiite commited on
Commit
36b23b2
·
1 Parent(s): d37628e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -13
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 = ['back_bumper',
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)