xiaolong0728 / YOLO

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Object detection for infrared images using YOLOv5

Dataset

  1. Flir
  2. Kaist

Improvement

  1. Add P2 layer
  2. Dilation and sub-pixel conv
  3. Add transformer mechanism
  4. Image denoising + enhancing
  5. YOLO + MAE

Training

This project runs on Google Colab

About


Languages

Language:Jupyter Notebook 95.1%Language:Python 4.8%Language:HTML 0.0%