cjf8899 / Pole_Transformer_Detection

:zap:Transformer detection on power poles

Pole_Transformer_Detection

This repo is a code that detects the transformer of the electric pole.

Demo

Getting Started

The pole detection model used YOLOv5 model.

Download models

• best.pt
  

the structures would like

Pole_Transformer_Detection/
    -- weights/
        -- best.pt
    -- videos/input/
        -- test.mp4
    -- detect.py
    ...

Constraints

• I can't reveal the dataset.:sweat_smile:
  
• We used a docker and the image we used was YOLOv5 image.
  

Demo Run

python detect.py

Referenced. Thank you all:+1:

detection code : https://github.com/ultralytics/yolov5