facebookresearch/pycls
Codebase for Image Classification Research, written in PyTorch.
| repo name | facebookresearch/pycls |
| repo link | https://github.com/facebookresearch/pycls |
| homepage | |
| language | Python |
| size (curr.) | 61 kB |
| stars (curr.) | 774 |
| created | 2019-06-10 |
| license | MIT License |
pycls
pycls is an image classification codebase, written in PyTorch. The codebase was originally developed for a project that led to the On Network Design Spaces for Visual Recognition work. pycls has since matured into a general image classification codebase that has been adopted by a number representation learning projects at Facebook AI Research.
Introduction
The goal of pycls is to provide a high-quality, high-performance codebase for image classification research. It is designed to be simple and flexible in order to support rapid implementation and evaluation of research ideas.
The codebase implements efficient single-machine multi-gpu training, powered by PyTorch distributed package. pycls includes implementations of standard baseline models (ResNet, ResNeXt, EfficientNet) and generic modeling functionality that can be useful for experimenting with network design. Additional models can be easily implemented.
Installation
Please see INSTALL.md for installation instructions.
Getting Started
After installation, please see GETTING_STARTED.md for basic instructions on training and evaluation with pycls.
Model Zoo
Coming soon!
Citing pycls
If you use pycls in your research, please use the following BibTex entry
@InProceedings{Radosavovic2019,
title = {On Network Design Spaces for Visual Recognition},
author = {Radosavovic, Ilija and Johnson, Justin and Xie, Saining and Lo, Wan-Yen and Doll{\'a}r, Piotr},
booktitle = {ICCV},
year = {2019},
}
License
pycls is released under the MIT license. See the LICENSE file for more information.
