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

catalyst-team/detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

Releases

No releases published

Packages

No packages published

Languages