Shypd.aiThe largest open index of AI toolsfpn.pytorch is an open-source Python library that provides a PyTorch implementation of Feature Pyramid Network (FPN) for object detection. It supports training with batch sizes greater than 1 and multiple GPUs.
Trending
About
fpn.pytorch offers a pure PyTorch implementation of the Feature Pyramid Network (FPN) for object detection, building upon the properties of a faster R-CNN implementation. This project stands out for its complete conversion of all NumPy implementations to PyTorch, ensuring a consistent and efficient environment. A key feature is its support for training with batch sizes greater than one, achieved by revising all relevant layers including dataloader, RPN, and ROI-pooling. It also leverages a multiple GPU wrapper (nn.DataParallel) for flexible scaling across one or more GPUs. The implementation integrates three pooling methods—ROI pooling, ROI align, and ROI crop—all adapted for multi-image batch training. Benchmarking has been conducted on datasets like PASCAL VOC and COCO, demonstrating its performance.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
Also listed in
