Skip to content

tensorlayer/TLXZoo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

96 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pre-trained models based on TensorLayerX. TensorLayerX is a multi-backend AI framework, which can run on almost all operation systems and AI hardwares, and support hybrid-framework programming. The currently version supports TensorFlow, MindSpore, PaddlePaddle, PyTorch, OneFlow and Jittor as the backends.

Quick Start

Installation

pip

# install from pypi
pip3 install tlxzoo

docker

docker pull xiaolong0612/tlxzoo:0.0.1
# run docker
cd /workspace/tlxzoo
python demo/vision/image_classification/vgg/predict.py

train

python demo/vision/image_classification/vgg/train.py

pretrained model

https://pan.baidu.com/s/1FZiHSthgX2FqynBP9cJ6_A?pwd=8dbs

predict

python demo/vision/image_classification/vgg/predict.py

Scenes and Models

TLXZOO currently covers 4 fields and 12 tasks. They ares listed below:

Vision

Object Detection

Image Classification

Image Segmentation

Face Recognition

Human Pose Estimation

Facial Landmark Detection

OCR

Text

Text Classification

Text Conditional Generation

Text Token Classidication

Speech

Automatic Speech Recognition

Graph

Node Classification