rcnn_inference_test.py 391 B

123456
  1. from watermark_verify.inference.rcnn import predict_and_detect
  2. if __name__ == '__main__':
  3. # detect_result = predict_and_detect('trigger/images/2/street.jpg', 'faster-rcnn.onnx', 'trigger/qrcode_positions.txt', (600, 600))
  4. detect_result = predict_and_detect('trigger/images/0/000000000030.jpg', 'faster-rcnn.onnx', 'trigger/qrcode_positions.txt', (600, 600))
  5. print(detect_result)