All Projects → AutoML-Research → AutoSTR

AutoML-Research / AutoSTR

Licence: other
H. Zhang, Q. Yao, M. Yang, Y. Xu, X. Bai. AutoSTR: Efficient Backbone Search for Scene Text Recognition. European Conference on Computer Vision (ECCV). 2020.

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to AutoSTR

Naszilla
Naszilla is a Python library for neural architecture search (NAS)
Stars: ✭ 181 (+135.06%)
Mutual labels:  neural-architecture-search
NASBench-PyTorch
A PyTorch implementation of NASBench
Stars: ✭ 43 (-44.16%)
Mutual labels:  neural-architecture-search
HyperKeras
An AutoDL tool for Neural Architecture Search and Hyperparameter Optimization on Tensorflow and Keras
Stars: ✭ 29 (-62.34%)
Mutual labels:  neural-architecture-search
Atomnas
Code for ICLR 2020 paper 'AtomNAS: Fine-Grained End-to-End Neural Architecture Search'
Stars: ✭ 197 (+155.84%)
Mutual labels:  neural-architecture-search
awesome-transformer-search
A curated list of awesome resources combining Transformers with Neural Architecture Search
Stars: ✭ 194 (+151.95%)
Mutual labels:  neural-architecture-search
MetaD2A
Official PyTorch implementation of "Rapid Neural Architecture Search by Learning to Generate Graphs from Datasets" (ICLR 2021)
Stars: ✭ 49 (-36.36%)
Mutual labels:  neural-architecture-search
Nsga Net
NSGA-Net, a Neural Architecture Search Algorithm
Stars: ✭ 171 (+122.08%)
Mutual labels:  neural-architecture-search
bristol
Parallel random matrix tools and complexity for deep learning
Stars: ✭ 25 (-67.53%)
Mutual labels:  neural-architecture-search
Auto-Compression
Automatic DNN compression tool with various model compression and neural architecture search techniques
Stars: ✭ 19 (-75.32%)
Mutual labels:  neural-architecture-search
torchprune
A research library for pytorch-based neural network pruning, compression, and more.
Stars: ✭ 133 (+72.73%)
Mutual labels:  neural-architecture-search
Awesome Nas Papers
Awesome Neural Architecture Search Papers
Stars: ✭ 213 (+176.62%)
Mutual labels:  neural-architecture-search
AutoSpeech
[InterSpeech 2020] "AutoSpeech: Neural Architecture Search for Speaker Recognition" by Shaojin Ding*, Tianlong Chen*, Xinyu Gong, Weiwei Zha, Zhangyang Wang
Stars: ✭ 195 (+153.25%)
Mutual labels:  neural-architecture-search
Hypernets
A General Automated Machine Learning framework to simplify the development of End-to-end AutoML toolkits in specific domains.
Stars: ✭ 221 (+187.01%)
Mutual labels:  neural-architecture-search
Hyperactive
A hyperparameter optimization and data collection toolbox for convenient and fast prototyping of machine-learning models.
Stars: ✭ 182 (+136.36%)
Mutual labels:  neural-architecture-search
receptive field analysis toolbox
A toolbox for receptive field analysis and visualizing neural network architectures
Stars: ✭ 84 (+9.09%)
Mutual labels:  neural-architecture-search
Lycoris
A lightweight and easy-to-use deep learning framework with neural architecture search.
Stars: ✭ 180 (+133.77%)
Mutual labels:  neural-architecture-search
MASTER-pytorch
Code for the paper "MASTER: Multi-Aspect Non-local Network for Scene Text Recognition" (Pattern Recognition 2021)
Stars: ✭ 263 (+241.56%)
Mutual labels:  scene-text-recognition
DUN
Code for "Depth Uncertainty in Neural Networks" (https://arxiv.org/abs/2006.08437)
Stars: ✭ 65 (-15.58%)
Mutual labels:  neural-architecture-search
CM-NAS
CM-NAS: Cross-Modality Neural Architecture Search for Visible-Infrared Person Re-Identification (ICCV2021)
Stars: ✭ 39 (-49.35%)
Mutual labels:  neural-architecture-search
onn
Online Deep Learning: Learning Deep Neural Networks on the Fly / Non-linear Contextual Bandit Algorithm (ONN_THS)
Stars: ✭ 139 (+80.52%)
Mutual labels:  neural-architecture-search

AutoSTR: Efficient Backbone Search for Scene Text Recognition

We investigate how to obtain a strong feature sequence extractor for scene text recognition task by neural architecture search technology. The research paper can be found here ECCV. 2020.

overview

Requirements

python==3.6.7
pytorch==1.4.0
torchvision==0.2.1
lmdb
PyYAML
pillow
editdistance
...

Searching Network Architecture

python3 arch_search_exp.py --config_file configs/search.yaml 

Retraining Compact Structure

python3 main.py --config_file configs/retrain.yaml 

logs and checkpoints

The logs and checkpoints can be found in here with extraction code wp8w.

Citation

If you find this work helpful for your research, please cite the following paper:

@inproceedings{zhang2020efficient,
  title={AutoSTR: Efficient Backbone Search for Scene Text Recognition},
  author={Zhang, Hui and Yao, Quanming and Yang, Mingkun and Xu, Yongchao and Bai, Xiang},
  booktitle={Proceedings of the European Conference on Computer Vision (ECCV)},
  year={2020}
}
@TechReport{yao2018taking,
  author      = {Yao, Quanming and Wang, Mengshuo},
  institution = {arXiv preprint},
  title       = {Taking Human out of Learning Applications: A Survey on Automated Machine Learning},
  year        = {2018},
}

Acknowledgement

We used the code part from aster.pytorch (https://github.com/ayumiymk/aster.pytorch) and proxylessnas(https://github.com/mit-han-lab/proxylessnas). Thanks for their excellent work very much.

New Opportunities

  • Interns, research assistants, and researcher positions are available. See requirement
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].