ytfeng commited on
Commit
3c4ef91
·
1 Parent(s): cc1f5f5

fix: set backend and target in init (#293)

Browse files
models/license_plate_detection_yunet/lpd_yunet.py CHANGED
@@ -21,6 +21,9 @@ class LPD_YuNet:
21
 
22
  # load model
23
  self.model = cv.dnn.readNet(self.model_path)
 
 
 
24
  # generate anchors/priorboxes
25
  self._priorGen()
26
 
 
21
 
22
  # load model
23
  self.model = cv.dnn.readNet(self.model_path)
24
+ # set backend and target
25
+ self.model.setPreferableBackend(self.backend_id)
26
+ self.model.setPreferableTarget(self.target_id)
27
  # generate anchors/priorboxes
28
  self._priorGen()
29