Skip to content

furkankupcu/RF-DETR_CROP_SINGLE_CLASS

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

14 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

RF-DETR Video Object Detection Tool

This project uses the RF-DETR model to detect specific objects in a video. It generates two outputs based on the detected objects:

  • πŸ”³ Cropped Video: A video composed only of cropped frames containing the detected objects.
  • πŸŸ₯ Annotated Video: The original video with bounding boxes drawn around detected objects.

πŸ› οΈ Requirements

You need the following Python packages installed:

pip install -r requirements.txt

python main.py
--video-path "videos/input_video.mp4"
--cropped-output-path "outputs/cropped_output.mp4"
--annotated-output-path "outputs/annotated_output.mp4"
--class-id 1
--threshold 0.5
--fps 25
--padding 20


Argument Description
--video-path Path to the input video file (required)
--cropped-output-path Output path for the cropped objects video
--annotated-output-path Output path for the annotated full-frame video
--class-id COCO class ID to detect (e.g. person = 1)
--threshold Confidence threshold for detection (default: 0.5)
--fps Frames per second for output videos
--padding Padding in pixels around bounding boxes when cropping
outputs/
β”œβ”€β”€ cropped_output.mp4       # Video with only cropped detected objects
└── annotated_output.mp4     # Original video with bounding boxes drawn

About

No description or website provided.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages