Showing 1-4 of 4 projects
A PyTorch library that implements various attention mechanisms and other neural network components for visual tasks.
A PyTorch implementation of the YOLOv4 object detection model with support for various attention mechanisms and mobile models.
Non-official implementation of the CBAM paper, a convolutional block attention module for neural networks.
A compression toolkit for YOLOv5 with support for various backbones, modules, and deployment options.
Get weekly updates on trending AI coding tools and projects.