lucasjinreal/pytorch_chatbot lucasjinreal/pytorch_chatbot

  • Stars
    star
    226
  • Rank 176,514 (Top 4 %)
  • Language
    Python
  • Created over 7 years ago
  • Updated over 5 years ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
lucasjinreal/pytorch_chatbot
github

lucasjinreal

Reviews

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

Repository Details

A Marvelous ChatBot implement using PyTorch.

PyTorch Marvelous ChatBot

[Update]

it's 2019 now, previously model can not catch up state-of-art now. So we just move towards the future a transformer based chatbot, it's much more accurate and flexiable as well as full of imagination interms of being a cahtbot!!! More importantly we are opensourced the whole codes here: http://manaai.cn/aicodes_detail3.html?id=36 Be sure to check it if you interested in chatbot and NLP!! it's build with tensorflow 2.0 newest api!!

Aim to build a Marvelous ChatBot

Synopsis

This is the first and the only opensource of ChatBot, I will continues update this repo personally, aim to build a intelligent ChatBot, as the next version of Jarvis.

This repo will maintain to build a Marvelous ChatBot based on PyTorch, welcome star and submit PR.

Already Done

Currently this repo did those work:

  • based on official tutorial, this repo will move on develop a seq2seq chatbot, QA system;
  • re-constructed whole project, separate mess code into data, model, train logic;
  • model can be save into local, and reload from previous saved dir, which is lack in official tutorial;
  • just replace the dataset you can train your own data!

Last but not least, this project will maintain or move on other repo in the future but we will continue implement a practical seq2seq based project to build anything you want: Translate Machine, ChatBot, QA System... anything you want.

Requirements

PyTorch
python3
Ubuntu Any Version
Both CPU and GPU can works

Usage

Before dive into this repo, you want to glance the whole structure, we have these setups:

  • config: contains the config params, which is global in this project, you can change a global param here;
  • datasets: contains data and data_loader, using your own dataset, you should implement your own data_loader but just a liitle change on this one;
  • models: contains seq2seq model definition;
  • utils: this folder is very helpful, it contains some code may help you get out of anoying things, such as save model, or catch KeyboardInterrupt exception or load previous model, all can be done in here.

to train model is also straightforward, just type:

python3 train.py

Contribute

wecome submit PR!!!! Let's build ChatBot together!

Contact

if you have anyquestion, you can find me via wechat jintianiloveu

More Repositories

1

tensorflow_poems

中文古诗自动作诗机器人，屌炸天，基于tensorflow1.10 api，正在积极维护升级中，快star，保持更新！
Python
3,595
star
2

yolov7_d2

🔥🔥🔥🔥 (Earlier YOLOv7 not official one) YOLO with Transformers and Instance Segmentation, with TensorRT acceleration! 🔥🔥🔥
Python
3,116
star
3

weibo_terminater

Final Weibo Crawler Scrap Anything From Weibo, comments, weibo contents, followers, anything. The Terminator
Python
2,303
star
4

alfred

alfred-py: A deep learning utility library for **human**, more detail about the usage of lib to: https://zhuanlan.zhihu.com/p/341446046
Python
854
star
5

DCNv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)
C++
564
star
6

keras_frcnn

Keras Implementation of faster-rcnn
Python
521
star
7

thor

thor: C++ helper library, for deep learning purpose
C++
264
star
8

tensorflow_novelist

模仿莎士比亚创作戏剧！屌炸天的是还能创作金庸武侠小说！快star，保持更新！！
Python
259
star
9

weibo_terminator_workflow

Update Version of weibo_terminator, This is Workflow Version aim at Get Job Done!
Python
258
star
10

faceswap_pytorch

Deep fake ready to train on any 2 pair dataset with higher resolution
Python
242
star
11

nb

Neural Network Blocks - Collect all kinds of fancy model blocks for you to build more powerful neural network model.
Python
231
star
12

CenterNet_Pro_Max

Experiments based on CenterNet (more backbones, TensorRT deployment and mask head)
221
star
13

LSTM_learn

a implement of LSTM using Keras for time series prediction regression problem
Python
214
star
14

AI-Infer-Engine-From-Zero

关于自建AI推理引擎的手册，从0开始你需要知道的所有事情
206
star
15

movenet

Google's Next Gen Pose Estimation in PyTorch
Python
122
star
16

Spider12306

基于Python3的12306抢票爬虫，10个线程开抢，智能过滤凌晨12：00到7：00发车的车次。
Python
106
star
17

pytorch_image_classifier

Minimal But Practical Image Classifier Pipline Using Pytorch, Finetune on ResNet18, Got 99% Accuracy on Own Small Datasets.
Python
106
star
18

kitti-ssd

Train your own data using SSD in a more clear and simple way(not include source code)
Python
101
star
19

TrafficLightsDetection

using SSD and caffe detect traffic lights on LISA dataset
Python
99
star
20

nosmpl

Accelerated SMPL operation, commonly used in generate 3D human mesh, STAR included.
Python
93
star
21

ssds_pytorch

Multiple basenet MobileNet v1,v2, ResNet combined with SSD detection method and it's variants such as RFB, FSSD etc.
Python
80
star
22

simpleocv

Make a minimal OpenCV runable on any where, WIP
C++
72
star
23

yolov3_tf2

Yolov3 implemented with brand new TensorFlow 2.0 API (both train and prediction)
Python
67
star
24

yolov7-face

Next Gen Face detection based on YOLOv7
Python
55
star
25

FruitsNutsSeg

detectron2 support self-define data train
Python
50
star
26

Q-Learning

An C++ Version of Q-Learning, to Train Robot Play with Flappybird!!
C++
40
star
27

cityscapestococo

This repo contains usable code convert cityscapes to coco format (Detectron and maskrcnn-benchmark were all broken)
Python
36
star
28

tfboys

TensorFlow and Pytorch practice codes with purity and simplicity.
Python
34
star
29

OpenHandMocap

Python
33
star
30

textfrontend

单独维护的中文TTS
Python
31
star
31

bboxer

Pure, Simple yet Powerful Image Bound Box Making Tool, already cross platform, welcome star and keep updating.
C++
31
star
32

fpn_rssd

Rotated Box SSD detection Framework with FPN support, next generation object detection framework
Python
29
star
33

aural

A Tiny Project For ASR model training and Deployment
Python
28
star
34

Shadowless

A Fast and Open Source Autonomous Perception System.
C++
27
star
35

awesome_transformer

A curated list of transformer learning materials, shared blogs, technical reviews.
26
star
36

pt_mobilenetv2_deeplabv3

Fast accurate realtime segmentation with DeepLabV3 and MobileNetV2 backbone
Python
26
star
37

pytorch_cycle_gan

CycleGAN with Productive Generate APIs. Generate Any Image from Your Transfer Model.
Python
26
star
38

yolovn

Just another yolo variant.
25
star
39

wanwu_release

Wanwu models release, code will be released soon
23
star
40

spconv

Pytorch layer needed by Second Lidar detector.
C++
23
star
41

3d_detection_kit

Toolkit to Explore 3D data for 3D object detection, point cloud visualization, bev map gen etc. Using KITTI as dummy data
Python
22
star
42

pytorch_image_caption

Image Caption, Show and Tell.
Python
20
star
43

datasets

A Collection of Datasets.
19
star
44

pilgrim_torch2trt

Pilgrim Project: torch2trt, quick convert your pytorch model to TensorRT engine.
C++
19
star
45

yolov5_mask

Try add Instance Segmentation upon YoloV5
Python
18
star
46

libnms

libnms.so for object detection, can be use in libtorch or caffe or nccn or onnx or TensorRT
Cuda
17
star
47

pt_enet

Realtime segmentation with ENet, the fast and accurate segmentation net.
Python
14
star
48

GreatDarkNet

An Edit Version of darknet, and this version you can train and predict on your own datasets! more easily!
C
14
star
49

VIBE_yolov5

Using YOLOv5 as detection on VIBE
Python
13
star
50

daybreak_release

Daybreak APP release
12
star
51

gofind

gofind - your personal find helper
Go
12
star
52

cabinet

Cabinet, The Ultimate Tool Box.
Rust
12
star
53

tensorflow_yolov3

A Detailed and Optimized Implementation of Yolo-V3 in Original TensorFlow.
Python
12
star
54

pytorch_name_net

A NetWork Generate Names, Based On Conditional RNN, Set Condition And Generate Different Names.
Python
11
star
55

tensorflow_extractor

State-of-art and Reliable Text-summary and Information Extraction
Python
11
star
56

RetinaNet

Pytorch Implementation of RetinaNet with CUDA accelerate nms operation.
Python
10
star
57

gluon_ssd

Implement SSD using Gluon in only 300 lines of codes!
Python
10
star
58

m

m editor is a modern, easy to use, fast terminal editor then vim or emacs. written in pure Rust.
Rust
10
star
59

wnnx_models

Various test models in WNNX format. It can view with `pip install wnetron && wnetron`
10
star
60

seg_icnet

ICNet in TensorFlow, Real-Time Segmentation
Python
10
star
61

scraper_toolbox

Python3.6 Scraper Toolbox, You can almost Scrap Anything in this Repo, Welcome Pull Request
Python
9
star
62

fusion

Fusion package with transformation between camera and lidar, IMU etc. Autonomous and robot helper.
Python
9
star
63

blackpearl

The Black Pearl in Golang. Personal Assistant.
Go
9
star
64

mxnet_tiny5

mxnet训练自己的数据集分类，支持模型断点训练和预测单张图片
Python
9
star
65

TTS_CN

A Chinese TTS System!
Python
9
star
66

arxiv_action

企业微信机器人或钉钉机器人定制服务，自动推送arxiv最新paper
Python
9
star
67

tacotron

TensorFlow implementation of Google Tacotron. Train on Audio and Generate Speech using Text. Which can be Called TTS.
Python
8
star
68

papers

Contains many papers with categories in CV, NLP, RL Quantum etc.
8
star
69

pytorch_style_transfer

A Simple Implementation of Neural Style Transfer using Pytorch. You can generate your own art pictures now!
Python
8
star
70

yolov8

7
star
71

numgo

NumPy library in Go.
Go
7
star
72

LLaVA-Magvit2

Python
7
star
73

PoseAILiveLink

PoseAI LiveLink Compatible on macOS
C++
7
star
74

gooooup

Upload load images(files) to cloud, generate permanent link.
Go
7
star
75

mjolnir

Light weighted replacement of original thor C++ library. More simpler, more clean, more light.
C++
7
star
76

UbuntuScripts

Shell
7
star
77

MLLM_Factory

A Dead Simple and Modularized Multi-Modal Training and Finetune Framework. Compatible to any LLaVA/Flamingo/QwenVL/MiniGemini etc series models.
7
star
78

visiontransformers

Vision Transformers that you need.
Python
6
star
79

sherpa_ort

ONNXRuntime ASR C++
C++
6
star
80

minitr

Exploration on Micro Transformers, Unleash the power of mini-transformers!
Python
6
star
81

tensorflow_wgan

A Tensorfow Version of the state-of-art Wasserstein GAN, image super resolution, black image colorful, more function are applying...just star!
Python
6
star
82

tensorflow_classifier

Simple and over-through process for Tensorflow classify images, using own dataset
Python
6
star
83

mxnet_ssd

Another maintained mxnet ssd version
Python
6
star
84

caffe_tiny5

Caffe tutorial for train own data and predict using python
Python
6
star
85

mmc

Next Gen MMD runs on all platforms, Windows, Linux, Mac. Will support exchange between vmd and fbx format.
C++
6
star
86

squeezeseg_pytorch

Realtime Point Cloud Segmentation
Python
6
star
87

vits_cpp

C++ and ONNXRuntime based VITS voice synthesis
C++
6
star
88

tf_pose_realtime

Realtime Openpose with MobileNetV2 backend
PureBasic
6
star
89

realrender

3D mesh render without pain.
C++
6
star
90

mono_odometry

Visual Odometry Using Mono Camera
C++
6
star
91

AwesomeLLM

6
star
92

person_tracking

person tracking in ros
C++
5
star
93

sparrow

The message server in Golang, like WeChat.
JavaScript
5
star
94

algorithm

Contains all kinds of algorithm write in Python and C++, some with Rust.
Python
5
star
95

CaffeHandsOn

This is a Caffe hands on tutorial.
Python
5
star
96

sak

Swiss Army Knife for secret hacking and sniffering
Go
5
star
97

instance_seg_tf

Instance Segmentation with discriminate loss
Python
5
star
98

efficientformers

Collection of efficient transformers.
Python
5
star
99

mumoda

Library to lean big models combined with Text and Image. And then Diffusion!
Python
5
star
100

picbed

Picbed utils for many open APIs
Go
4
star