Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

PyTorch

PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab.

https://github.com/kuan-wang/pytorch-mobilenet-v3

MobileNetV3 in pytorch and ImageNet pretrained models

classification imagenet mobilenet mobilenetv2 mobilenetv3 pytorch

Last synced: 31 Oct 2024

https://github.com/Spijkervet/SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.

contrastive-learning pytorch representation-learning simclr unsupervised-learning

Last synced: 05 Nov 2024

https://github.com/kaituoxu/Speech-Transformer

A PyTorch implementation of Speech Transformer, an End-to-End ASR with Transformer network on Mandarin Chinese.

asr attention attention-is-all-you-need end-to-end pytorch self-attention transformer

Last synced: 07 Aug 2024

https://github.com/higgsfield/Capsule-Network-Tutorial

Pytorch easy-to-follow Capsule Network tutorial

capsule-network clean-code easy-to-use pytorch pytorch-tutorials

Last synced: 28 Oct 2024

https://github.com/spijkervet/simclr

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.

contrastive-learning pytorch representation-learning simclr unsupervised-learning

Last synced: 11 Oct 2024

https://github.com/shibing624/similarities

Similarities: a toolkit for similarity calculation and semantic search. 相似度计算、匹配搜索工具包,支持亿级数据文搜文、文搜图、图搜图,python3开发,开箱即用。

bm25 deep-learning faiss image-search image-similarity matching nlp pytorch search-engine similarity similarity-search text-matching

Last synced: 22 Oct 2024

https://github.com/Denis2054/Transformers-for-NLP-2nd-Edition

Transformer models from BERT to GPT-4, environments from Hugging Face to OpenAI. Fine-tuning, training, and prompt engineering examples. A bonus section with ChatGPT, GPT-3.5-turbo, GPT-4, and DALL-E including jump starting GPT-4, speech-to-text, text-to-speech, text to image generation with DALL-E, Google Cloud AI,HuggingGPT, and more

bert chatgpt chatgpt-api dall-e dall-e-api deep-learning gpt-3-5-turbo gpt-4 gpt-4-api huggingface-transformers machine-learning natural-language-processing nlp openai python pytorch roberta-model transformers trax

Last synced: 24 Oct 2024

https://github.com/xinntao/facexlib

FaceXlib aims at providing ready-to-use face-related functions based on current STOA open-source methods.

alignment assessment deep-learning detection face headpose matting parsing pytorch recognition tracking

Last synced: 22 Oct 2024

https://github.com/positive666/yolo_research

based on yolo-high-level project (detect\pose\classify\segment\):include yolov5\yolov7\yolov8\ core ,improvement research ,SwintransformV2 and Attention Series. training skills, business customization, engineering deployment C

attention-mechanism custom-networ deepstream object-detection pytorch swintransformer yolov5 yolov7 yolov8

Last synced: 09 Nov 2024

https://github.com/rusty1s/pytorch_cluster

PyTorch Extension Library of Optimized Graph Cluster Algorithms

cluster-algorithms geometric-deep-learning graph-neural-networks pytorch

Last synced: 22 Oct 2024

https://github.com/SimonVandenhende/Multi-Task-Learning-PyTorch

PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).

computer-vision eccv2020 multi-task-learning nyud pascal pytorch scene-understanding segmentation

Last synced: 02 Aug 2024

https://github.com/fastnlp/fasthan

fastHan是基于fastNLP与pytorch实现的中文自然语言处理工具,像spacy一样调用方便。

bert cws fastnlp joint-model ner parser pos python pytorch

Last synced: 05 Nov 2024

https://github.com/hengyuan-hu/bottom-up-attention-vqa

An efficient PyTorch implementation of the winning entry of the 2017 VQA Challenge.

bottom-up-attention pytorch vqa

Last synced: 07 Nov 2024

https://github.com/teddykoker/torchsort

Fast, differentiable sorting and ranking in PyTorch

cuda-kernel pytorch ranking sort

Last synced: 03 Aug 2024

https://github.com/wmcnally/kapao

KAPAO is an efficient single-stage human pose estimation model that detects keypoints and poses as objects and fuses the detections to predict human poses.

deep-learning human-pose-estimation pose-estimation pytorch yolo

Last synced: 09 Nov 2024

https://github.com/fastnlp/fastHan

fastHan是基于fastNLP与pytorch实现的中文自然语言处理工具,像spacy一样调用方便。

bert cws fastnlp joint-model ner parser pos python pytorch

Last synced: 27 Oct 2024

https://github.com/fregu856/deeplabv3

PyTorch implementation of DeepLabV3, trained on the Cityscapes dataset.

autonomous-driving computer-vision deep-learning machine-learning pytorch semantic-segmentation

Last synced: 07 Aug 2024

https://github.com/asyml/texar-pytorch

Integrating the Best of TF into PyTorch, for Machine Learning, Natural Language Processing, and Text Generation. This is part of the CASL project: http://casl-project.ai/

bert casl-project data-processing deep-learning dialog-systems gpt-2 machine-learning machine-translation natural-language-processing python pytorch roberta texar texar-pytorch text-data text-generation xlnet

Last synced: 02 Aug 2024

https://github.com/Tencent/PatrickStar

PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.

bert gpt nlp pretrained-models pytorch

Last synced: 03 Aug 2024

https://github.com/guochengqian/PointNeXt

[NeurIPS'22] PointNeXt: Revisiting PointNet++ with Improved Training and Scaling Strategies

3d-vision pointcloud pointnet2 pytorch

Last synced: 28 Oct 2024

https://github.com/soapisnotfat/super-resolution

collection of super-resolution models & algorithms

fsrcnn pytorch resolution-models srcnn srgan sup-pixel-cnn

Last synced: 05 Aug 2024

https://github.com/msnh2012/Msnhnet

🔥 (yolov3 yolov4 yolov5 unet ...)A mini pytorch inference framework which inspired from darknet.

darknet inference-engine jetson-nx mobilenetv2 mobilenetyolo pytorch yolov3 yolov4 yolov5

Last synced: 09 Nov 2024

https://github.com/chenyuntc/pytorch-best-practice

A Guidance on PyTorch Coding Style Based on Kaggle Dogs vs. Cats

image-classification pytorch visdom

Last synced: 30 Oct 2024

https://github.com/williamFalcon/test-tube

Python library to easily log experiments and parallelize hyperparameter search for neural networks

caffe caffe2 chainer data-science deep-learning grid-search hyperparameter-optimization keras machine-learning neural-networks pytorch random-search tensorflow

Last synced: 30 Oct 2024

https://github.com/williamfalcon/test-tube

Python library to easily log experiments and parallelize hyperparameter search for neural networks

caffe caffe2 chainer data-science deep-learning grid-search hyperparameter-optimization keras machine-learning neural-networks pytorch random-search tensorflow

Last synced: 30 Oct 2024

https://github.com/jackroos/VL-BERT

Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".

bert iclr2020 pre-training pytorch representation-learning self-supervised-learning vision-and-language vl-bert

Last synced: 03 Nov 2024

https://github.com/kymatio/kymatio

Wavelet scattering transforms in Python with GPU acceleration

deep-learning keras numpy pytorch scattering-transform scikit-learn tensorflow wavelets

Last synced: 10 Oct 2024

https://github.com/fboulnois/stable-diffusion-docker

Run the official Stable Diffusion releases in a Docker container with txt2img, img2img, depth2img, pix2pix, upscale4x, and inpaint.

dall-e dalle diffusion docker generative-art huggingface image-generation inpainting midjourney pytorch stable-diffusion tensorflow text-to-image

Last synced: 30 Oct 2024

https://github.com/allentdan/libtorchtutorials

This is a code repository for pytorch c++ (or libtorch) tutorial.

classification cpp libtorch pytorch pytorch-cpp resnet segmentation tutorial-code unet vgg yolov4 yolov4-tiny

Last synced: 01 Nov 2024

https://github.com/PaddlePaddle/X2Paddle

Deep learning model converter for PaddlePaddle. (『飞桨』深度学习模型转换工具)

caffe model-converter onnx paddlepaddle pytorch tensorflow x2paddle-model-zoo

Last synced: 06 Nov 2024

https://github.com/paddlepaddle/x2paddle

Deep learning model converter for PaddlePaddle. (『飞桨』深度学习模型转换工具)

caffe model-converter onnx paddlepaddle pytorch tensorflow x2paddle-model-zoo

Last synced: 06 Nov 2024

https://github.com/chungyiweng/humannerf

HumanNeRF turns a monocular video of moving people into a 360 free-viewpoint video.

cvpr2022 humannerf nerf pytorch

Last synced: 07 Nov 2024

https://github.com/luhua-rain/MRC_Competition_Dureader

机器阅读理解 冠军/亚军代码及中文预训练MRC模型

bert dureader mrc pytorch qa squad

Last synced: 02 Nov 2024

https://github.com/quic/sense

Enhance your application with the ability to see and interact with humans using any RGB camera.

activity-recognition calorie-estimation computer-vision deep-learning fitness-app gesture-recognition neural-networks pytorch video

Last synced: 05 Aug 2024

https://github.com/drsleep/light-weight-refinenet

Light-Weight RefineNet for Real-Time Semantic Segmentation

deep-learning light-weight pytorch refinenet semantic-segmentation

Last synced: 09 Nov 2024

https://github.com/MisaOgura/flashtorch

Visualization toolkit for neural networks in PyTorch! Demo -->

cnn deep-learning explainability interpretability machine-learning neural-networks pytorch visualization

Last synced: 30 Oct 2024

https://github.com/DrSleep/light-weight-refinenet

Light-Weight RefineNet for Real-Time Semantic Segmentation

deep-learning light-weight pytorch refinenet semantic-segmentation

Last synced: 04 Aug 2024

https://github.com/ritchieng/deep-learning-wizard

Open source guides/codes for mastering deep learning to deploying deep learning in production in PyTorch, Python, Apptainer, and more.

apptainer deep-learning llamaindex ollama python3 pytorch

Last synced: 10 Oct 2024

https://github.com/creafz/pytorch-cnn-finetune

Fine-tune pretrained Convolutional Neural Networks with PyTorch

convolutional-neural-networks deep-learning pretrained-models pytorch

Last synced: 30 Oct 2024

https://github.com/microsoft/Tutel

Tutel MoE: An Optimized Mixture-of-Experts Implementation

mixture-of-experts moe nlp pytorch transformer

Last synced: 01 Nov 2024

https://github.com/hkchengrex/Cutie

[CVPR 2024 Highlight] Putting the Object Back Into Video Object Segmentation

computer-vision cvpr2024 deep-learning pytorch segmentation video-editing video-object-segmentation video-segmentation

Last synced: 08 Nov 2024

https://github.com/hkchengrex/cutie

[CVPR 2024 Highlight] Putting the Object Back Into Video Object Segmentation

computer-vision cvpr2024 deep-learning pytorch segmentation video-editing video-object-segmentation video-segmentation

Last synced: 10 Nov 2024

https://github.com/mhamilton723/stego

Unsupervised Semantic Segmentation by Distilling Feature Correspondences

computer-vision deep-learning iclr2022 pytorch semantic-segmentation unsupervised-learning

Last synced: 01 Nov 2024

https://github.com/cadene/vqa.pytorch

Visual Question Answering in Pytorch

clevr coco deep-learning pytorch resnet skipthoughts torch vgenome vqa

Last synced: 07 Nov 2024

https://github.com/microsoft/tutel

Tutel MoE: An Optimized Mixture-of-Experts Implementation

mixture-of-experts moe nlp pytorch transformer

Last synced: 07 Oct 2024

https://github.com/zhunzhong07/Random-Erasing

Random Erasing Data Augmentation. Experiments on CIFAR10, CIFAR100 and Fashion-MNIST

aaai2020 data-augmentation image-classification object-detection person-re-identification pytorch

Last synced: 02 Aug 2024

https://github.com/microsoft/CameraTraps

PyTorch Wildlife: a Collaborative Deep Learning Framework for Conservation.

camera-traps computer-vision conservation machine-learning megadetector pytorch pytorch-wildlife wildlife

Last synced: 08 Aug 2024

https://github.com/jia-zhuang/pytorch-multi-gpu-training

整理 pytorch 单机多 GPU 训练方法与原理

multi-gpu-training pytorch

Last synced: 03 Nov 2024

https://github.com/weiaicunzai/Bag_of_Tricks_for_Image_Classification_with_Convolutional_Neural_Networks

experiments on Paper <Bag of Tricks for Image Classification with Convolutional Neural Networks> and other useful tricks to improve CNN acc

image-classification pytorch

Last synced: 04 Nov 2024

https://github.com/Cadene/vqa.pytorch

Visual Question Answering in Pytorch

clevr coco deep-learning pytorch resnet skipthoughts torch vgenome vqa

Last synced: 02 Nov 2024

https://github.com/lonepatient/albert_pytorch

A Lite Bert For Self-Supervised Learning Language Representations

albert bert language-model mask ngram nlp pytorch

Last synced: 06 Nov 2024

https://github.com/lzx1413/PytorchSSD

pytorch version of SSD and it's enhanced methods such as RFBSSD,FSSD and RefineDet

fssd pytorch refinedet rfb ssd

Last synced: 03 Aug 2024

https://github.com/run-house/runhouse

The fastest way to iterate and deploy AI workloads on your own infra. Unobtrusive, debuggable, PyTorch-like APIs.

api artificial-intelligence aws azure collaboration data-science deployment distributed fastapi gcp infrastructure machine-learning middleware observability python pytorch ray sagemaker serverless

Last synced: 11 Oct 2024

https://github.com/lonePatient/albert_pytorch

A Lite Bert For Self-Supervised Learning Language Representations

albert bert language-model mask ngram nlp pytorch

Last synced: 02 Nov 2024

https://github.com/onepanelio/onepanel

The open source, end-to-end computer vision platform. Label, build, train, tune, deploy and automate in a unified platform that runs on any cloud and on-premises.

ai aiops annotation computer-vision deeplearning etl hyperparameter-tuning inference jupyterlab labeling machinelearning mlops pipelines pytorch tensorboard tensorflow training workflows

Last synced: 05 Nov 2024

https://github.com/SMILELab-FL/FedLab

A flexible Federated Learning Framework based on PyTorch, simplifying your Federated Learning research.

deep-learning federated-learning federated-learning-framework fedlab machine-learning pytorch pytorch-federated-learning

Last synced: 02 Aug 2024

https://github.com/zhiqwang/yolort

yolort is a runtime stack for yolov5 on specialized accelerators such as tensorrt, libtorch, onnxruntime, tvm and ncnn.

deployment detection graghsurgeon inference jit libtorch lightning ncnn nms onnx onnxruntime pytorch tensorrt torchscript trt tvm visualization yolo yolort yolov5

Last synced: 28 Oct 2024

https://github.com/ethanhe42/kl-loss

Bounding Box Regression with Uncertainty for Accurate Object Detection (CVPR'19)

detection detection-algorithm detection-model detection-network object-detection pytorch

Last synced: 01 Nov 2024

https://github.com/mosaicml/streaming

A Data Streaming Library for Efficient Neural Network Training

dataset deep-learning machine-learning neural-network pytorch streaming

Last synced: 11 Nov 2024

https://github.com/yl-1993/learn-to-cluster

Learning to Cluster Faces (CVPR 2019, CVPR 2020)

cvpr2019 cvpr2020 face-clustering face-recognition gcn pytorch

Last synced: 02 Nov 2024

https://github.com/fidler-lab/polyrnn-pp-pytorch

PyTorch training/tool code for Polygon-RNN++ (CVPR 2018)

annotation cvpr2018 deep-learning instance-segmentation labelling polygon-rnn pytorch

Last synced: 07 Aug 2024

https://github.com/bobbens/sketch_simplification

Models and code related to sketch simplification of rough sketches.

convolutional-neural-networks deep-learning pytorch siggraph sketch torch

Last synced: 09 Nov 2024

https://github.com/stevenygd/PointFlow

PointFlow : 3D Point Cloud Generation with Continuous Normalizing Flows

3d-point-clouds computer-vision continuous-normalizing-flows machine-learning pytorch shapes

Last synced: 05 Aug 2024

https://github.com/wohlert/semi-supervised-pytorch

Implementations of various VAE-based semi-supervised and generative models in PyTorch

generative-models pytorch semi-supervised-learning

Last synced: 03 Aug 2024

https://github.com/alfredxiangwu/lightcnn

A Light CNN for Deep Face Representation with Noisy Labels, TIFS 2018

face-recognition lightcnn pytorch

Last synced: 09 Nov 2024

https://github.com/ShichenLiu/CondenseNet

CondenseNet: Light weighted CNN for mobile devices

deep-learning mobile-device pytorch

Last synced: 31 Oct 2024

https://github.com/jonasrauber/eagerpy

PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code

eager-execution jax numpy python pytorch tensorflow tensorflow2

Last synced: 15 Oct 2024

https://github.com/shichenliu/condensenet

CondenseNet: Light weighted CNN for mobile devices

deep-learning mobile-device pytorch

Last synced: 06 Nov 2024

https://github.com/jayleicn/ClipBERT

[CVPR 2021 Best Student Paper Honorable Mention, Oral] Official PyTorch code for ClipBERT, an efficient framework for end-to-end learning on image-text and video-text tasks.

cvpr2021 pytorch video-question-answering video-retrieval vision-and-language vqa

Last synced: 03 Aug 2024

https://github.com/csteinmetz1/auraloss

Collection of audio-focused loss functions in PyTorch

audio loss-functions pytorch

Last synced: 05 Aug 2024

https://github.com/HumanCompatibleAI/overcooked_ai

A benchmark environment for fully cooperative human-AI performance.

artificial-intelligence deep-learning machine-learning pytorch reinforcement-learning

Last synced: 10 Nov 2024

https://github.com/mhamilton723/STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences

computer-vision deep-learning iclr2022 pytorch semantic-segmentation unsupervised-learning

Last synced: 03 Aug 2024

https://github.com/ibm/ffdl

Fabric for Deep Learning (FfDL, pronounced fiddle) is a Deep Learning Platform offering TensorFlow, Caffe, PyTorch etc. as a Service on Kubernetes

ai artificial-intelligence caffe deep-learning deep-neural-networks deeplearning ibm-research-ai jupyter keras kubernetes-cluster machine-learning ml model python pytorch storage tensorflow watson

Last synced: 29 Oct 2024

https://github.com/humancompatibleai/overcooked_ai

A benchmark environment for fully cooperative human-AI performance.

artificial-intelligence deep-learning machine-learning pytorch reinforcement-learning

Last synced: 01 Oct 2024

https://github.com/thunlp/openattack

An Open-Source Package for Textual Adversarial Attack.

adversarial-attacks adversarial-example natural-language-processing nlp pytorch

Last synced: 10 Nov 2024

https://github.com/wkentaro/pytorch-for-numpy-users

PyTorch for Numpy users. https://pytorch-for-numpy-users.wkentaro.com

numpy pytorch

Last synced: 30 Oct 2024

https://github.com/haofanwang/controlnet-for-diffusers

Transfer the ControlNet with any basemodel in diffusers🔥

aigc controlnet diffusers pytorch tutorials

Last synced: 07 Nov 2024

https://github.com/open-mmlab/mmfewshot

OpenMMLab FewShot Learning Toolbox and Benchmark

few-shot-learning openmmlab pytorch

Last synced: 11 Nov 2024

https://github.com/antao97/dgcnn.pytorch

A PyTorch implementation of Dynamic Graph CNN for Learning on Point Clouds (DGCNN)

classification point-cloud pytorch segmentation

Last synced: 28 Oct 2024

https://github.com/apple/ml-aim

This repository provides the code and model checkpoints of the research paper: Scalable Pre-training of Large Autoregressive Image Models

jax large-scale-vision-models mlx pytorch

Last synced: 07 Oct 2024