8000 GitHub - sailfish009/pycls: Codebase for Image Classification Research, written in PyTorch.
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Codebase for Image Classification Research, written in PyTorch.

License

Notifications You must be signed in to change notification settings

sailfish009/pycls

 
 

Repository files navigation

pycls

pycl 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.

About

Codebase for Image Classification Research, written in PyTorch.

Resources

License

Code of conduct

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.8%
  • Makefile 0.2%
0