Skip to content
프로젝트 카테고리

머신러닝 · GitHub

모델 학습, 평가, 추론 프레임워크를 찾고 커뮤니티의 관심도를 비교하세요.

유용한 GitHub 프로젝트

7,586 프로젝트

tiatoolbox

TissueImageAnalytics

568스타

Computational Pathology Toolbox developed by TIA Centre, University of Warwick.

EPro-PnP

tjiiv-cprg

1,177스타
EPro-PnP preview

[CVPR 2022 Best Student Paper] EPro-PnP: Generalized End-to-End Probabilistic Perspective-n-Points for Monocular Object Pose Estimation

머신러닝PythonApache-2.0GitHub
1,486스타
GANotebooks preview

wgan, wgan2(improved, gp), infogan, and dcgan implementation in lasagne, keras, pytorch

머신러닝Jupyter NotebookMITGitHub

gcn

tkipf

7,417스타
gcn preview

Implementation of Graph Convolutional Networks in TensorFlow

58스타
iris-3b-metal preview

Iris-3B pixel-space text-to-image on Apple Silicon GPUs: Metal-optimized engine, native macOS app, CLI, Python and local HTTP API

머신러닝PythonApache-2.0GitHub
571스타
fire-detection-cnn preview

real-time fire detection in video imagery using a convolutional neural network (deep learning) - from our ICIP 2018 paper (Dunnings / Breckon) + ICMLA 2019 paper (Samarth / Bhowmik / Breckon)

1,385스타

This note presents in a technical though hopefully pedagogical way the three most common forms of neural network architectures: Feedforward, Convolutional and Recurrent.

skrl

Toni-SM

1,104스타

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments

TorchDR

TorchDR

217스타
TorchDR preview

TorchDR - PyTorch Dimensionality Reduction

  • High Performance - Engineered for speed with GPU acceleration, torch.compile support, and optimized algorithms leveraging sparsity and negative sampling.
  • Multi-GPU Support - Scale to massive datasets with built-in distributed computing. Use the torchdr CLI or torchrun for easy multi-GPU execution of compatible modules and methods.
머신러닝PythonBSD-3-ClauseGitHub

Ensemble-Pytorch

TorchEnsemble-Community

1,038스타

A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.

머신러닝PythonBSD-3-ClauseGitHub

torchgeo

torchgeo

4,245스타
torchgeo preview

TorchGeo: 지리공간 데이터를 위한 데이터셋, 샘플러, 변환 및 사전 학습 모델

tsl

TorchSpatiotemporal

392스타

tsl: a PyTorch library for processing spatiotemporal data.

  • Create Custom Models and Datasets Easily build your own custom models and datasets for spatiotemporal data analysis. Whether you're working with sensor networks, environmental data, or any other spatiotemporal domain, tsl's high-level APIs empower you to develop tailored solutions.
  • Access a Wealth of Existing Datasets and Models Leverage a vast collection of datasets and models from the spatiotemporal data processing literature. Explore and benchmark against state-of-the-art baselines, and test your brand new model on widely used public datasets.