Ask HN: How to train an image recognition AI

• detectron2

  49 28,744 7.6 Python

  Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

  

I don’t do AI professionally but as a hobby, so this may not be the best way. But the way you described, it seems the user maybe taking the picture a bit further away and there may be other objects in the frame. So you may want to look into some sort of segmentation or have bounding box. This could help the user make sure they are looking at documents for the correct machine.

I think something like detectron2 [1] could help. It is Apache2 license, so commercial friendly. That said the pre-trained weights may not be used for commercial purposes, so you’ll want to check on that.

[1] https://github.com/facebookresearch/detectron2

• WorkOS

  workos.com featured

  The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

  

Suggest a related project