/YOYOF-in-defect-detection

The first to adopt the YOLOF model on a metal defect NEU-DET dataset. Using several image preprocessing methods, including Edge Detection, Denoising, Sharpening, and SR, were experimented on the metal defect NEU-DET dataset. In the proposed optimized batch size, backbone, soft efficient non-maximum sup-pression (Soft-NMS), and threshold of YOLO.

Primary LanguageJupyter Notebook

Watchers