Implementation of ['YOLO-OB: An improved anchor-free real-time multi-scale colon polyp detector in colonoscopy']
Published: 29 August 2023| Version 1 | DOI: 10.17632/xj3x7vmsdg.1
Contributor:
sean yanDescription
This repo is the implementation of ['YOLO-OB: An improved anchor-free real-time multi-scale colon polyp detector in colonoscopy'] The deep learning-based object detection algorithm is currently one of the most effective methods for improving polyp detection during colonoscopy. This repository proposes a new model called YOLO-OB based on YOLOv3. It improves network's feature fusion capacity by adopting the Spatial Pyramid Pooling (SPP) and the Bi-directional Feature Pyramid Network (BiFPN) modules. It also employs the ObjectBox detection head, which uses a center-based anchor-free box regression strategy that can detect polyps of different sizes on feature maps of any scale.
Files
Steps to reproduce
Follow the readme document.
Institutions
Huazhong University of Science and Technology School of Computer Science and Technology
Categories
Object Detection Algorithm