Skip to content

KUD-00/segment-cv

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Segmentation

this repo provides

  • python scripts
  • .pt model

showcase

track detect segmentation trajectory segment-trajectory

models

  • car detection model: link
  • large vehicle head and body detection model: not uploaded

install dependencies

  • use requirement.txt ( only way now )
  • use shell.nix( without ultralytics package, can't do models things )

scripts

clip-video.py

  • extract a specific video clip
  • Usage: python3 clip-video.py <video path> <start time> <end time> <output directory>

get-frame.py

  • get a specific frame picture from video(or process a folder of video)
  • Usage: python3 get-frame.py <video_path>/<video_folder_path> <time_str> <output_path>
  • time_str can be exactly '3:42' or vague '1/2'

predict-image.py

  • predict objects in image
  • Usage: python3 predict-picture.py <image-path> <model-path>

track.py

  • do tracking
  • Usage: python3 track.py <model path> <source video>

train.py

  • train models
  • Usage: python3 train.py <model path> <yaml path> <epoches>

train-resume.py

  • resume train models
  • Usage: python3 train-resume.py <model path>

extract-car-image.py

  • extract all single car images from a whole image
  • Usage: python3 extract-car-image.py <model path> <image path> <output dir path> <scale factor>

everything in utils is for specific things, not for general

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published