gmarti commited on
Commit
6801764
1 Parent(s): 2d7d4a4

modify app

Browse files
Files changed (1) hide show
  1. app.py +1 -12
app.py CHANGED
@@ -5,12 +5,9 @@ import yolov5
5
  import cv2
6
  import torch
7
  import numpy as np
8
- from models.experimental import attempt_load
9
 
10
  # load model
11
- #model_face = torch.hub.load('.', 'custom', '/home/gmarti/Documentos/detection/yolov5s-face.pt', source='local', force_reload=True).to("cpu")
12
- #torch.save(model_face, 'face_model.pt')
13
- model_face = torch.load('/home/gmarti/Documentos/detection/face_model.pt')
14
  model_face.eval()
15
  model_plate = yolov5.load('keremberke/yolov5m-license-plate', device="cpu")
16
  # # set model parameters
@@ -21,13 +18,6 @@ model_plate.multi_label = False # NMS multiple labels per box
21
  model_plate.max_det = 1000 # maximum number of detections per image
22
 
23
 
24
- #model = torch.hub.load('.', 'yolov5s','/home/gmarti/Documentos/detection/yolov5s-face.pt', 'yolov5s', device='cpu', source='local')
25
-
26
- #model_face = torch.load('/home/gmarti/Documentos/detection/yolov5s-face.pt', map_location=torch.device("cpu"))
27
- #model_face = attempt_load('/home/gmarti/Documentos/detection/yolov5s-face.pt', map_location=torch.device("cpu"))
28
- #model_face.to(torch.device("cpu"))
29
-
30
-
31
  def blur_plates_image(image, plate_blur):
32
 
33
  results_plate = model_plate(image, augment=True)
@@ -46,7 +36,6 @@ def blur_plates_image(image, plate_blur):
46
  def blur_faces_image(image, face_blur):
47
 
48
  results = model_face(np.array(image))
49
- #results = model_face(torch.from_numpy(image))
50
 
51
  # parse results
52
  boxes_face_list = results.pred[0][:, :4].cpu().numpy().tolist() # x1, y1, x2, y2
 
5
  import cv2
6
  import torch
7
  import numpy as np
 
8
 
9
  # load model
10
+ model_face = torch.load('face_model.pt')
 
 
11
  model_face.eval()
12
  model_plate = yolov5.load('keremberke/yolov5m-license-plate', device="cpu")
13
  # # set model parameters
 
18
  model_plate.max_det = 1000 # maximum number of detections per image
19
 
20
 
 
 
 
 
 
 
 
21
  def blur_plates_image(image, plate_blur):
22
 
23
  results_plate = model_plate(image, augment=True)
 
36
  def blur_faces_image(image, face_blur):
37
 
38
  results = model_face(np.array(image))
 
39
 
40
  # parse results
41
  boxes_face_list = results.pred[0][:, :4].cpu().numpy().tolist() # x1, y1, x2, y2