PRBNet_PyTorch

KO | EN

gitlite — search

#javascript #python #react #typescript #go #android #machine-learning #rust #macos #java #nodejs #windows #security #ios

(★ 74)

No description available.

PRBNet_PyTorch Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# PRBNet PyTorch
This is the reference PyTorch implementation for training and testing  [single-shot object detection](https://github.com/pingyang1117/PRBNet_PyTorch/tree/main/prb) and [oriented bounding boxes](https://github.com/pingyang1117/PRBNet_PyTorch/tree/main/obb) models using the method described in

> **[Parallel Residual Bi-Fusion Feature Pyramid Network for Accurate Single-Shot Object Detection](https://ieeexplore.ieee.org/abstract/document/9603994)**
>
> [Ping-Yang, Chen](https://scholar.google.com.tw/citations?user=cGHFHm0AAAAJ&hl=zh-TW), [Ming-Ching Chang](https://www.albany.edu/computer-science/faculty/ming-ching-chang), [Jun-Wei Hsieh](https://aicvlab2019.wordpress.com/), and [Yong-Sheng Chen](https://www.cs.nycu.edu.tw/members/detail/yschen)
>
> [TIP 2021 (arXiv pdf)](https://arxiv.org/abs/2012.01724)

## Performance

### MS COCO
#### P5 Model

| Model | Test Size | APtest | AP50test | AP75test | APstest | FPS |
| :-- | :-: | :-: | :-: | :-: | :-: | :-: | 
| **YOLOv7-x** | 640 | **53.1%** | **71.2%** | **57.8%** | **33.8%** | **114** | 
| | | | | | | | 
| [**PRB-FPN-CSP**](http://140.113.110.150:5000/sharing/AZc2PgpWN) | 640 | **51.8%** | **70.0%** | **56.7%** | **32.6%** | **113** | 
| [**PRB-FPN-MSP**](http://140.113.110.150:5000/sharing/Es4M7Vprv) | 640 | **53.3%** | **71.1%** | **58.3%** | **34.1%** | **94** | 
| [**PRB-FPN-ELAN**](http://140.113.110.150:5000/sharing/orXcxcGSw) | 640 | **52.5%** | **70.4%** | **57.2%** | **33.4%** | **70** | 
| | | | | | | |

#### P6 Model
| Model | Test Size | APtest | AP50test | AP75test | FPS |
| :-- | :-: | :-: | :-: | :-: | :-: | 
| **YOLOv7-E6E** | 1280 | **56.8%** | **74.4%** | **62.1%** | **36**| 
| | | | | | | | 
| [**PRB-FPN6**](http://140.113.110.150:5000/sharing/xvwUNT3zZ) | 1280 | **56.9%** | **74.1%** | **62.3%** | **31**| 
| [**PRB-FPN6-MSP**](http://140.113.110.150:5000/sharing/wavtpaPzu) | 1280 | **57.2%** | **74.5%** | **62.5%** | **27**| 
| | | | | | | |

If you find our work useful in your research please consider citing our paper:

```
@ARTICLE{9603994,
  author={Chen, Ping-Yang and Chang, Ming-Ching and Hsieh, Jun-Wei and Chen, Yong-Sheng},
  journal={IEEE Transactions on Image Processing}, 
  title={Parallel Residual Bi-Fusion Feature Pyramid Network for Accurate Single-Shot Object Detection}, 
  year={2021},
  volume={30},
  number={},
  pages={9099-9111},
  doi={10.1109/TIP.2021.3118953}}
```

If you find the backbone also well-done in your research, please consider citing the CSPNet. Most of the credit goes to Dr. Wang:

```
@inproceedings{wang2020cspnet,
  title={{CSPNet}: A New Backbone That Can Enhance Learning Capability of {CNN}},
  author={Wang, Chien-Yao and Mark Liao, Hong-Yuan and Wu, Yueh-Hua and Chen, Ping-Yang and Hsieh, Jun-Wei and Yeh, I-Hau},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops},
  pages={390--391},
  year={2020}
}
```

# Acknowledgement
Without the guidance of Dr. Mark Liao and a discussion with Dr. Wang, PRBNet would not have been published quickly in TIP and open-sourced to the community. Many of the code is borrowed from [YOLOv4](https://github.com/WongKinYiu/PyTorch_YOLOv4), [YOLOv5_obb](https://github.com/hukaixuan19970627/yolov5_obb), and [YOLOv7](https://github.com/WongKinYiu/yolov7). Many thanks for their fantastic work:

* https://github.com/AlexeyAB/darknet
* https://github.com/ultralytics/yolov3
* https://github.com/WongKinYiu/PyTorch_YOLOv4
* https://github.com/ultralytics/yolov5
* https://github.com/hukaixuan19970627/yolov5_obb
* https://github.com/WongKinYiu/yolov7

PRBNet_PyTorch

PRBNet_PyTorch Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

PRBNet_PyTorch

PRBNet_PyTorch Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README