Jittor-Image-Models / Jittor-Image-Models

Jittor Image Models is a library for pulling together a wide variety of SOTA deep learning models in the Jittor framework.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Jittor Image Models

Jittor Image Models (jimm) is a library for pulling together a wide variety of SOTA deep learning models in the Jittor framework. Based on jimm, we achieved the first place of the Dog Species Classification track in the Jittor AI Competition in 2021.

Our jimm is modified from PyTorch Image Models (timm) which helps fine-tune PyTorch models list systematically by timm in Jittor.

More specifically, PyTorch Image Models (timm) is an excellent project created by Ross Wightman and perfected by many outstanding contributors. Details about timm is available at: https://github.com/rwightman/pytorch-image-models

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators. More details about Jittor can be found via: https://github.com/Jittor/jittor

In our jimm, we reproduce part of pytorch-image-models in the Jittor deep learning framework, and also provide a training demo to make it easier for you to get started.

Update News

Feb 25, 2022

Sep 06, 2021

June 10, 2021

  • Add HRNet models.

Currently Supported Models

More models provided by timm will continue to be updated.

Results

Model validation results can be found in the following url: https://rwightman.github.io/pytorch-image-models/results/

TODO List

Contacts

If you have any questions about our work, please do not hesitate to contact us by emails.

Xuhao Sun: sunxh@njust.edu.cn

Yang Shen: shenyang_98@njust.edu.cn

Xiu-Shen Wei (Primary contact): weixs.gm@gmail.com

About

Jittor Image Models is a library for pulling together a wide variety of SOTA deep learning models in the Jittor framework.


Languages

Language:Python 100.0%