YOLOv4 and Enhanced SRGAN

Example of Input Image

Example of Detection Image

Example of cropped Image

Example of final Image

Trained Models

Use gdown to download the trained model from our GitHub:

gdown https://github.com/RuiyangJu/YOLO_SR/releases/download/Trained_Model/Trained_Model.zip

Usage

• YOLOv4 pretrained models in [Yolo_SR/model_data]
• ESRGAN pretrained models in [Yolo_SR/models]
• Object Detection

    python3 yolo_test.py

• Super-resolution

    python3 SR_test.py

References

• Cropped Detection YOLOv4
• Super-Resolution ESRGAN