get_res.py 888 B

12345678910111213141516171819202122232425262728293031
  1. import ddddocr
  2. import base64
  3. def get_res(x):
  4. ocr = ddddocr.DdddOcr(beta=True)
  5. if x.get('jqImage') and x.get('syImage'):
  6. jqImage=base64.b64decode(x.get('jqImage'))
  7. jqImageCode = ocr.classification(jqImage)
  8. syImage = base64.b64decode(x.get('syImage'))
  9. syImageCode = ocr.classification(syImage)
  10. return {
  11. "jqImageCode": jqImageCode,
  12. "syImageCode": syImageCode
  13. }
  14. elif x.get('jqImage') and x.get('syImage')=='':
  15. jqImage = base64.b64decode(x.get('jqImage'))
  16. jqImageCode = ocr.classification(jqImage)
  17. return {
  18. "jqImageCode": jqImageCode,
  19. }
  20. elif x.get('jqImage')=='' and x.get('syImage'):
  21. syImage = base64.b64decode(x.get('syImage'))
  22. syImageCode = ocr.classification(syImage)
  23. return {
  24. "syImageCode": syImageCode,
  25. }