Wanli commited on
Commit
fefbdcf
·
1 Parent(s): 284d26a

fix bugs in wechatQRcode (#153)

Browse files
models/qrcode_wechatqrcode/demo.py CHANGED
@@ -28,13 +28,13 @@ parser = argparse.ArgumentParser(
28
  description="WeChat QR code detector for detecting and parsing QR code (https://github.com/opencv/opencv_contrib/tree/master/modules/wechat_qrcode)")
29
  parser.add_argument('--input', '-i', type=str,
30
  help='Usage: Set path to the input image. Omit for using default camera.')
31
- parser.add_argument('--detect_prototxt_path', type=str, default='detect_2021sep.prototxt',
32
  help='Usage: Set path to detect.prototxt.')
33
- parser.add_argument('--detect_model_path', type=str, default='detect_2021sep.caffemodel',
34
  help='Usage: Set path to detect.caffemodel.')
35
- parser.add_argument('--sr_prototxt_path', type=str, default='sr_2021sep.prototxt',
36
  help='Usage: Set path to sr.prototxt.')
37
- parser.add_argument('--sr_model_path', type=str, default='sr_2021sep.caffemodel',
38
  help='Usage: Set path to sr.caffemodel.')
39
  parser.add_argument('--backend_target', '-bt', type=int, default=0,
40
  help='''Choose one of the backend-target pair to run this demo:
 
28
  description="WeChat QR code detector for detecting and parsing QR code (https://github.com/opencv/opencv_contrib/tree/master/modules/wechat_qrcode)")
29
  parser.add_argument('--input', '-i', type=str,
30
  help='Usage: Set path to the input image. Omit for using default camera.')
31
+ parser.add_argument('--detect_prototxt_path', type=str, default='detect_2021nov.prototxt',
32
  help='Usage: Set path to detect.prototxt.')
33
+ parser.add_argument('--detect_model_path', type=str, default='detect_2021nov.caffemodel',
34
  help='Usage: Set path to detect.caffemodel.')
35
+ parser.add_argument('--sr_prototxt_path', type=str, default='sr_2021nov.prototxt',
36
  help='Usage: Set path to sr.prototxt.')
37
+ parser.add_argument('--sr_model_path', type=str, default='sr_2021nov.caffemodel',
38
  help='Usage: Set path to sr.caffemodel.')
39
  parser.add_argument('--backend_target', '-bt', type=int, default=0,
40
  help='''Choose one of the backend-target pair to run this demo:
models/qrcode_wechatqrcode/wechatqrcode.py CHANGED
@@ -15,7 +15,7 @@ class WeChatQRCode:
15
  sr_prototxt_path,
16
  sr_model_path
17
  )
18
- if backendId != 0:
19
  raise NotImplementedError("Backend {} is not supported by cv.wechat_qrcode_WeChatQRCode()".format(backendId))
20
  if targetId != 0:
21
  raise NotImplementedError("Target {} is not supported by cv.wechat_qrcode_WeChatQRCode()")
@@ -25,7 +25,7 @@ class WeChatQRCode:
25
  return self.__class__.__name__
26
 
27
  def setBackendAndTarget(self, backendId, targetId):
28
- if backendId != 0:
29
  raise NotImplementedError("Backend {} is not supported by cv.wechat_qrcode_WeChatQRCode()".format(backendId))
30
  if targetId != 0:
31
  raise NotImplementedError("Target {} is not supported by cv.wechat_qrcode_WeChatQRCode()")
 
15
  sr_prototxt_path,
16
  sr_model_path
17
  )
18
+ if backendId != 0 and backendId != 3:
19
  raise NotImplementedError("Backend {} is not supported by cv.wechat_qrcode_WeChatQRCode()".format(backendId))
20
  if targetId != 0:
21
  raise NotImplementedError("Target {} is not supported by cv.wechat_qrcode_WeChatQRCode()")
 
25
  return self.__class__.__name__
26
 
27
  def setBackendAndTarget(self, backendId, targetId):
28
+ if backendId != 0 and backendId != 3:
29
  raise NotImplementedError("Backend {} is not supported by cv.wechat_qrcode_WeChatQRCode()".format(backendId))
30
  if targetId != 0:
31
  raise NotImplementedError("Target {} is not supported by cv.wechat_qrcode_WeChatQRCode()")