Skip to content
This repository has been archived by the owner on Oct 25, 2021. It is now read-only.

catalyst-team/detection

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
src
 
 
 
 
 
 
 
 
 
 
 
 

Catalyst logo

Accelerated DL R&D

Build Status CodeFactor Pipi version Docs PyPI Status

Twitter Telegram Slack Github contributors

PyTorch framework for Deep Learning research and development. It was developed with a focus on reproducibility, fast experimentation and code/ideas reusing. Being able to research/develop something new, rather than write another regular train loop.
Break the cycle - use the Catalyst!

Project manifest. Part of PyTorch Ecosystem. Part of Catalyst Ecosystem:

  • Alchemy - Experiments logging & visualization
  • Catalyst - Accelerated Deep Learning Research and Development
  • Reaction - Convenient Deep Learning models serving

Catalyst at AI Landscape.


Catalyst.Detection Build Status Github contributors

Note: this repo uses advanced Catalyst Config API and could be a bit out-of-day right now. Use Catalyst's minimal examples section for a starting point and up-to-day use cases, please.

Based on Objects as points article by Xingyi Zhou, Dequan Wang, Philipp Krähenbühl

Training in your dataset

  1. Install requirements pip install -r requirements.txt

  2. Copy all images to one directory or two different directories for train and validation.

  3. Create markup_train.json as json file in MSCOCO format using COCODetectionFactory from data_preparation.py. This class may be copied to your dataset generator. See documentation in code comments. If your dataset are already in this format, go to next step.

  4. Specify perameters and in config/centernet_detection_config.yml.

  5. Run catalyst catalyst-dl run --config=./configs/centernet_detection_config.yml

  6. When you change dataset, you must delete cache files markup_*.json.cache because this files contain preprocessed bounding boxes info.

About

Catalyst.Detection

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages