Update app.py
Browse files
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(
|
38 |
return "Can not detect license plate from image"
|
39 |
else:
|
40 |
system.eval()
|
41 |
pred_labels = []
|
42 |
-
for i in range(len(
|
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)
|