• Stars
    star
    189
  • Rank 204,649 (Top 5 %)
  • Language
    Python
  • Created about 5 years ago
  • Updated almost 4 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

cnn+rnn+attention: vgg(vgg16,vgg19)+rnn(LSTM, GRU)+attention, resnet(resnet_v2_50,resnet_v2_101,resnet_v2_152)+rnnrnn(LSTM, GRU)+attention, inception_v4+rnn(LSTM, GRU)+attention, inception_resnet_v2+rnn(LSTM, GRU)+attention,..... vgg(vgg16,vgg19), resnet(resnet_v2_50,resnet_v2_101,resnet_v2_152), inception_v4, inception_resnet_v2,.....

此repo是tensorflow版本的,pytorch版本:https://github.com/MachineLP/PyTorch_image_classifier

tensorflow搭建的一个训练框架,包含模型有:vgg(vgg16,vgg19), resnet(resnet_v2_50,resnet_v2_101,resnet_v2_152), inception_v4, inception_resnet_v2等。

此框架主要针对分类任务, 后面会陆续搭建多任务多标签、检测、以及rnn等框架,欢迎关注。 搭建时使用的环境为:Python3.5, tensorflow1.4

具体:

实现基础cnn训练,数据读取方式慢。

优化数据读取的方式,学习率加入衰减。

在train_cnn_v0基础上加入rnn。

在train_cnn_v0基础上加入rnn、attention。

使用多GPU训练(默认两块gpu),以上其他框架使用多GPU,只需把train.py替换掉就可以了。

多任务多标签训练及其总结。

GANs训练及其总结。

理论及其代码实践。

MachineLP的日常代码。

对dl感兴趣,还可以关注我的博客,这是我的博客目录:(地址: http://blog.csdn.net/u014365862/article/details/78422372 ) 本文为博主原创文章,未经博主允许不得转载。有问题可以加微信:lp9628(注明CSDN)。

公众号MachineLN,邀请您扫码关注:

image

More Repositories

1

Tensorflow-

Tensorflow实战学习笔记、代码、机器学习进阶系列
Jupyter Notebook
1,129
star
2

TextMatch

QAmatch(qa_match)/文本匹配/文本分类/文本embedding/文本聚类/文本检索(bow/ifidf/ngramtf-df/bert/albert/bm25/…/nn/gbdt/xgb/kmeans/dscan/faiss/….)
Python
839
star
3

CodeFun

DataStructure(SwordOffer、LeetCode)、Deep Learning(Tensorflow、Keras、Pytorch)、Machine Learning(sklearn、spark)、AutoML、AutoDL、ModelDeploying、SQL
Jupyter Notebook
344
star
4

PyTorch_image_classifier

Image classification: efficientnet/resnest/seresnext/.....
Python
212
star
5

OCR_Keras

OCR;文本检测、文本识别(cnn+ctc、crnn+ctc)。
Python
208
star
6

OpenCV-

OpenCV学习笔记
C++
95
star
7

TensorFlowTTS_chinese

chinese tts
Jupyter Notebook
76
star
8

train_cnn-rnn

cnn+rnn: vgg(vgg16,vgg19)+rnn(LSTM, GRU), resnet(resnet_v2_50,resnet_v2_101,resnet_v2_152)+rnnrnn(LSTM, GRU), inception_v4+rnn(LSTM, GRU), inception_resnet_v2+rnn(LSTM, GRU),.....
Python
63
star
9

QDServing

ml模型分布式服务部署:grpc,flask;docker
Python
62
star
10

Spark-

Spark学习笔记
Jupyter Notebook
42
star
11

face_stickers

人脸贴纸
Python
32
star
12

train_cnn_GANs

GANs:DCGAN
Python
21
star
13

Pytorch_multi_task_classifier

multi-task classifier
Python
21
star
14

py_workSpace

MachineLP的工具包
Python
18
star
15

conditional-similarity-networks-Tensorflow

Conditional Similarity Networks (CSNs-Tensorflow)
Python
11
star
16

cnn_faceDetection

cpp code: using cnn for face detection
C++
7
star
17

LaneDetection

lane Detection
C++
5
star
18

qa-corpus-zh

问答语料库整理
5
star
19

FashionAI_KeyPoints_Detection

This my project of Fashion AI Key points detection contest.
Python
5
star
20

PyTorch_video_classifier

video recognition
Python
3
star
21

MachineLP

2
star
22

vehicleDetection

C++
2
star
23

Inception_v4_slim

【Ubuntu】slim框架下的inception_v4模型的运行、可视化、导出和使用
Python
1
star