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 yan

Description

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

Licence