PDG commited on
Commit
fb8c16f
·
1 Parent(s): e8d20c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -65,13 +65,13 @@ carTransforms = transforms.Compose([transforms.Resize(224),
65
  transforms.Normalize(mean=MEAN, std=STD)])
66
 
67
  def classifyCar(im):
68
- im = Image.fromarray(im.astype('uint8'), 'RGB')
69
- im = transforms.ToTensor(im)
70
  try:
71
  with torch.no_grad():
72
  outputs = predictor(im)
73
  except:
74
- return im, {"error1": im.shape}
75
  try:
76
  v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
77
  except Exception as err:
 
65
  transforms.Normalize(mean=MEAN, std=STD)])
66
 
67
  def classifyCar(im):
68
+ #im = Image.fromarray(im.astype('uint8'), 'RGB')
69
+ im = cv2.imread(im)
70
  try:
71
  with torch.no_grad():
72
  outputs = predictor(im)
73
  except:
74
+ return im, {"error1": 0.5}
75
  try:
76
  v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
77
  except Exception as err: