Skip to content
Project categories

Machine Learning · GitHub

Train, evaluate, and run machine learning models.

Useful GitHub projects

8,771 projects

DN4

WenbinLee

201stars

Pytorch code of "Revisiting Local Descriptor based Image-to-Class Measure for Few-shot Learning", CVPR 2019.

220stars

Data stream analytics: Implement online learning methods to address concept drift and model drift in data streams using the River library. Code for the paper entitled "PWPAE: An Ensemble Framework for Concept Drift Adaptation in IoT Data Streams" published in IEEE GlobeCom 2021.

Machine LearningJupyter NotebookMITGitHub

detect-waste

wimlds-trojmiasto

225stars
detect-waste preview

AI to Combat Environmental Pollution - detecting plastic waste in the environment to combat environmental pollution and promote circular economy (Deep Learning, PyTorch)

vitamin-cuda

WingEdge777

234stars

🍎 One kernel a day keeps high latency away. A hands-on CUDA learning path featuring a rich collection of kernels, from the basics to peak performance, seamlessly integrated as PyTorch C++ extensions.

fcn

wkentaro

215stars

Chainer Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)

FJSP-DRL

wrqccc

203stars

This repository is the official implementation of the paper “Flexible Job Shop Scheduling via Dual Attention Network Based Reinforcement Learning”. IEEE Transactions on Neural Networks and Learning Systems, 2023.

SED

xb534

202stars

[TPAMI2025&CVPR2024] Official Pytorch Implementation of SED: A Simple Encoder-Decoder for Open-Vocabulary Semantic Segmentation.

Machine LearningPythonApache-2.0GitHub

Stable-Makeup

Xiaojiu-z

234stars

Pytorch Implementation of "Stable-Makeup: When Real-World Makeup Transfer Meets Diffusion Model" (SIGGRAPH 2025)

Machine LearningPythonApache-2.0GitHub