windy2612 commited on
Commit
6c3ee18
·
verified ·
1 Parent(s): 1041d2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,12 +34,12 @@ def predict(image):
34
  if isinstance(image, str):
35
  image = cv2.imread(image)
36
  _, img_wapred, _, _ = detect_lp(wpod_net, image, 0.5)
37
- if len(img_warped) == 0:
38
  return "Can not detect license plate from image"
39
  else:
40
  system.eval()
41
  pred_labels = []
42
- for i in range(len(img_warped)):
43
  img = (img_wapred[i] * 255).astype(np.uint8)
44
  img = Image.fromarray(img).convert("RGB")
45
  image = trans(img).unsqueeze(0)
 
34
  if isinstance(image, str):
35
  image = cv2.imread(image)
36
  _, img_wapred, _, _ = detect_lp(wpod_net, image, 0.5)
37
+ if len(img_wapred) == 0:
38
  return "Can not detect license plate from image"
39
  else:
40
  system.eval()
41
  pred_labels = []
42
+ for i in range(len(img_wapred)):
43
  img = (img_wapred[i] * 255).astype(np.uint8)
44
  img = Image.fromarray(img).convert("RGB")
45
  image = trans(img).unsqueeze(0)