Patch models bugs - Githubissues

visual-layer / fastdup

fastdup is a powerful, free tool designed to rapidly generate valuable insights from image and video datasets. It helps enhance the quality of both images and labels, while significantly reducing data operation costs, all with unmatched scalability.

Other

1.59k stars 77 forks source link

Patch models bugs #293

Closed dnth closed 11 months ago

dnth commented 11 months ago

This PR makes patches to the zero-shot models.

Use cv2.cvtColor(img, cv2.COLOR_BGR2RGB) instead of img[:, :, ::-1] to convert image into to RGB.
Support loading custom weights for SAM.
Wrap SAM inference call in a try-catch block to catch errors.