Skip to content

Try quickly to find the right model and generate visualizations

Notifications You must be signed in to change notification settings

EasonEdison/detectron2-train-and-visual

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Detection Tool

Goal of this repo

if you want to use a model in detectron2 as front end to detect object in your dataset,you can use this repo to quickly try many model while don't neet to change config.yaml to fit your dataset,only change some parameter in train.py, and you can get visual result usevisual_2.py

How to use

Only need to change some parameter in train.py

files set

  1. put config file in config
  2. put pretrained weight in pretrain
  3. visual_v1.py and evalution use same weight path

change parameter

  1. change dataset path in root_davis
  2. when use visual_v1,recomment to set gpu=1
  3. if want to get else result ,such as daivs_test,change visual_what in visual_result.py

visual_v1 and visual_v2

visual_v1:

  • launch visual_v1 in train.py
  • need configandweight

visual_v2:

  • independent run,change parameter
  • need result.jason
  • need register dateset

About

Try quickly to find the right model and generate visualizations

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages