Update app.py
Browse files
app.py
CHANGED
|
@@ -9,8 +9,10 @@ import torch
|
|
| 9 |
import pathlib
|
| 10 |
import numpy as np
|
| 11 |
from urllib.parse import urlparse
|
| 12 |
-
from PIL import Image
|
| 13 |
from transformers import AutoImageProcessor, YolosForObjectDetection, DetrForObjectDetection
|
|
|
|
|
|
|
|
|
|
| 14 |
from PIL import Image, ImageDraw
|
| 15 |
from transformers import YolosImageProcessor, YolosForObjectDetection
|
| 16 |
import easyocr
|
|
@@ -68,6 +70,7 @@ model.eval()
|
|
| 68 |
|
| 69 |
reader = easyocr.Reader(['en'], gpu=False)
|
| 70 |
|
|
|
|
| 71 |
# -------------------- Plate Color Classifier --------------------
|
| 72 |
def classify_plate_color(plate_img):
|
| 73 |
img = np.array(plate_img)
|
|
|
|
| 9 |
import pathlib
|
| 10 |
import numpy as np
|
| 11 |
from urllib.parse import urlparse
|
|
|
|
| 12 |
from transformers import AutoImageProcessor, YolosForObjectDetection, DetrForObjectDetection
|
| 13 |
+
import sqlite3
|
| 14 |
+
import pandas as pd
|
| 15 |
+
import matplotlib.pyplot as plt
|
| 16 |
from PIL import Image, ImageDraw
|
| 17 |
from transformers import YolosImageProcessor, YolosForObjectDetection
|
| 18 |
import easyocr
|
|
|
|
| 70 |
|
| 71 |
reader = easyocr.Reader(['en'], gpu=False)
|
| 72 |
|
| 73 |
+
|
| 74 |
# -------------------- Plate Color Classifier --------------------
|
| 75 |
def classify_plate_color(plate_img):
|
| 76 |
img = np.array(plate_img)
|