機械学習 · GitHub

Pytorch model to caffe model, supported pytorch 0.3, 0.3.1, 0.4, 0.4.1 ,1.0 , 1.0.1 , 1.2 ,1.3 .notice that only pytorch 1.1 have some bugs

• Converting a pytorch model to caffe model.
• Some convenient tools of manipulate caffemodel and prototxt quickly(like get or set weights of layers).

(ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"

PyTorch Implementation of MVSNet

[ICLR 2024] Official Implementation of "Diffusion-TS: Interpretable Diffusion for General Time Series Generation"

A stock trading bot that uses machine learning to make price predictions.

Label-Efficient Semantic Segmentation with Diffusion Models (ICLR'2022)

[ICML 2023] The official implementation of the paper "TabDDPM: Modelling Tabular Data with Diffusion Models"

Faster Recurrent Neural Network Language Modeling Toolkit with Noise Contrastive Estimation and Hierarchical Softmax

Machine Learning toolbox for Humans

Paper list about multimodal and large language models, only used to record papers I read in the daily arxiv for personal needs.

Awesome Artificial Intelligence Projects

Everybody dance now reproduced in pytorch

Pytorch Implement of Dynamic Snake Convolution (ICCV2023)

TensorFlow and PyTorch implementation of "Meta-Transfer Learning for Few-Shot Learning" (CVPR2019)

Experiments used in "Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning"

"Pop Music Transformer: Beat-based Modeling and Generation of Expressive Pop Piano Compositions", ACM Multimedia 2020

[ICLR 2024] Official pytorch implementation of "ControlVideo: Training-free Controllable Text-to-Video Generation"

[COLING 2022] CSL: A Large-scale Chinese Scientific Literature Dataset 中文科学文献数据集

A mobilenet SSD based face detector, powered by tensorflow object detection api, trained by WIDERFACE dataset.