Skip to content
Project categories

Machine Learning · GitHub

Train, evaluate, and run machine learning models.

Useful GitHub projects

4,624 projects
2,534stars
UGATIT-pytorch preview

Official PyTorch implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation

5,208stars
watermark-removal preview

a machine learning image inpainting task that instinctively removes watermarks from image indistinguishable from the ground truth image

yolov3-tf2

zzh8829

2,513stars
yolov3-tf2 preview

YoloV3 Implemented in Tensorflow 2.0

  • yolov3 with pre-trained Weights
  • yolov3-tiny with pre-trained Weights
Machine LearningJupyter NotebookMITGitHub
48stars
Plant-Disease-Classification preview

A ResNet50-DCBAM deep learning framework for plant disease classification using the PlantVillage dataset with reproducible training and evaluation pipelines.

  • ResNet50 backbone with pretrained ImageNet weights
  • Channel attention

xlearn

aksnzhy

3,090stars
xlearn preview

High performance, easy-to-use, and scalable machine learning (ML) package, including linear model (LR), factorization machines (FM), and field-aware factorization machines (FFM) for Python and CLI interface.