SBDE: A Segmentation-driven Editing Method for Bolt Defect Augmentation and Detection

Official repository for SBDE: A Segmentation-driven Editing Method for Bolt Defect Augmentation and Detection

Updates

• Nov 29, 2024: ✨ You can refer to the Hugging Face model demo built by @xiaoyangjie for easier usage via this link.
• Nov 28, 2024: ✨ The code has been released!

[Hugging Face demo] (Recommended).

This example shows the test results of our method on a bolt image (Fig.2) selected from the public paper.

Requirements

*Python 3.8

*PyTorch 2.4.1

Todo

This code combines different tasks and is quite complex. We will organize this work in the future.

Acknowledgements

This code refers to the projects: RobustSAM, SAM-Adapter and LaMa Thanks for their excellent work！