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/snap-research/CAT

[CVPR 2021] Teachers Do More Than Teach: Compressing Image-to-Image Models (CAT)

compression cyclegan deep-learning gan gaugan image-to-image pix2pix pytorch

Last synced: 08 Aug 2024

https://github.com/coqui-ai/Trainer

🐸 - A general purpose model trainer, as flexible as it gets

ai data-science deep-learning machine-learning pytorch

Last synced: 07 Aug 2024

https://github.com/willbrennan/semanticsegmentation

A framework for training segmentation models in pytorch on labelme annotations with pretrained examples of skin, cat, and pizza topping segmentation

birds bisenet bisenetv2 cats coco computer-vision labelme labelme-annotations pizza pizza-toppings pytorch segmentation semantic-segmentation skin-detection skin-segmentation torchvision

Last synced: 27 Oct 2024

https://github.com/ChrisWaites/pyvacy

Differentially Private Optimization for PyTorch 👁🙅‍♀️

differential-privacy pytorch

Last synced: 02 Nov 2024

https://github.com/tjuskyzhang/Scaled-YOLOv4-TensorRT

Got 100fps on TX2. Got 500fps on GeForce GTX 1660 Ti. If the project is useful to you, please Star it.

pytorch scaled-yolov4 scaled-yolov4-tenosrrt tensorrt tx2 yolo yolov4-csp yolov4-csp-tensorrt yolov4-large yolov4-large-tensorrt yolov4-tiny yolov4-tiny-tensorrt

Last synced: 09 Nov 2024

https://github.com/BMW-InnovationLab/BMW-Anonymization-API

This repository allows you to anonymize sensitive information in images/videos. The solution is fully compatible with the DL-based training/inference solutions that we already published/will publish for Object Detection and Semantic Segmentation.

anonymization-api anonymization-technique bmw computer-vision data-anonymization deep-learning docker image-transformation inference no-code object-detection openvino privacy-enhancing-technologies privacy-protection pytorch semantic-segmentation tensorflow tensorflow-training video video-anonymization

Last synced: 05 Nov 2024

https://github.com/fedml-ai/fedgraphnn

FedGraphNN: A Federated Learning Platform for Graph Neural Networks with MLOps Support. The previous research version is accepted to ICLR'2021 - DPML and MLSys'21 - GNNSys workshops.

deep-learning distributed-learning federated-learning federated-learning-framework fedml gnns graph-neural-networks machine-learning pytorch tensorflow

Last synced: 08 Nov 2024

https://github.com/BMW-InnovationLab/BMW-Anonymization-API?referral=top-free-anonymization-tools-apis-and-open-source-models

This repository allows you to anonymize sensitive information in images/videos. The solution is fully compatible with the DL-based training/inference solutions that we already published/will publish for Object Detection and Semantic Segmentation.

anonymization-api anonymization-technique bmw computer-vision data-anonymization deep-learning docker image-transformation inference no-code object-detection openvino privacy-enhancing-technologies privacy-protection pytorch semantic-segmentation tensorflow tensorflow-training video video-anonymization

Last synced: 12 Aug 2024

https://github.com/david-knigge/ccnn

Code repository of the paper "Modelling Long Range Dependencies in ND: From Task-Specific to a General Purpose CNN" https://arxiv.org/abs/2301.10540.

convolutional-neural-networks deep-learning pytorch

Last synced: 28 Oct 2024

https://github.com/HennyJie/BrainGB

Officially Accepted to IEEE Transactions on Medical Imaging (TMI, IF: 11.037) - Special Issue on Geometric Deep Learning in Medical Imaging.

ai4health brain-networks gcn graph-convolutional-networks graph-neural-networks network-embedding pytorch

Last synced: 12 Nov 2024

https://github.com/Ai-trainee/Traffic-Sign-Recognition-PyQt5-YOLOv5-GUI

This YOLOv5🚀😊 GUI road sign system uses MySQL💽, PyQt5🎨, PyTorch, CSS🌈. It has modules for login🔑, YOLOv5 setup📋, sign recognition🔍, database💾, and image processing🖼️. It supports diverse inputs, model switching, and enhancements like mosaic and mixup📈.

css mysql-database pyqt5 pytorch tensorrt-engine yolov5 yolov5-gui

Last synced: 09 Nov 2024

https://github.com/yerfor/syntaspeech

SyntaSpeech: Syntax-aware Generative Adversarial Text-to-Speech; IJCAI 2022; Official code

gan pytorch speech-synthesis tts

Last synced: 06 Nov 2024

https://github.com/acecoooool/dss-pytorch

:star: PyTorch implement of Deeply Supervised Salient Object Detection with Short Connection

python pytorch salient-object-detection

Last synced: 09 Nov 2024

https://github.com/d-li14/psconv

[ECCV 2020] PSConv: Squeezing Feature Pyramid into One Compact Poly-Scale Convolutional Layer

convolution eccv2020 feature-pyramids instance-segmentation mmdetection multi-scale object-detection pytorch

Last synced: 07 Nov 2024

https://github.com/csyanbin/TPN-pytorch

Pytorch Code for ICLR19 paper: Learning to Propagate Labels: Transductive Propagation Network for Few-shot Learning.

few-shot-learning label-propagation meta-learning pytorch transductive-few-shot-learning

Last synced: 26 Oct 2024

https://github.com/eriklindernoren/Action-Recognition

Exploration of different solutions to action recognition in video, using neural networks implemented in PyTorch.

action-recognition pytorch video-classification

Last synced: 06 Nov 2024

https://github.com/d-li14/PSConv

[ECCV 2020] PSConv: Squeezing Feature Pyramid into One Compact Poly-Scale Convolutional Layer

convolution eccv2020 feature-pyramids instance-segmentation mmdetection multi-scale object-detection pytorch

Last synced: 03 Aug 2024

https://github.com/dansuh17/alexnet-pytorch

Pytorch Implementation of AlexNet

alexnet dataset paper pytorch

Last synced: 31 Oct 2024

https://github.com/pqhieu/jsis3d

[CVPR'19] JSIS3D: Joint Semantic-Instance Segmentation of 3D Point Clouds

cvpr deep-learning instance-segmentation point-cloud pytorch

Last synced: 28 Oct 2024

https://github.com/xiaoyu258/GeoProj

Blind Geometric Distortion Correction on Images Through Deep Learning

deep-learning distortion-correction pytorch

Last synced: 25 Oct 2024

https://github.com/phamquiluan/PubLayNet

ICDAR 2019: MaskRCNN on PubLayNet datasets. Paragraph detection, table detection, figure detection,...

document-layout-analysis figure-detection mask-rcnn object-detection paragraph-detection pretrained-models publaynet pytorch table-detection

Last synced: 08 Aug 2024

https://github.com/jzlianglu/pykaldi2

Yet another speech toolkit based on Kaldi and PyTorch

horovod kaldi pykaldi pytorch speech-toolkit

Last synced: 17 Aug 2024

https://github.com/recursionpharma/gflownet

GFlowNet library specialized for graph & molecular data

deep-learning gflownet graph-neural-network pytorch

Last synced: 03 Aug 2024

https://github.com/rayhane-mamah/efficient-vdvae

Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"

image-generation jax paper python pytorch representation-learning unsupervised-learning vae

Last synced: 06 Nov 2024

https://github.com/dengbocong/text-similarity

文本相似度(匹配)计算,提供Baseline、训练、推理、指标分析...代码包含TensorFlow/Pytorch双版本

bert deep-learning mechine-learing model nlp pytorch similarity text-classification transformer

Last synced: 08 Nov 2024

https://github.com/Pilhyeon/BaSNet-pytorch

Official Pytorch Implementation of 'Background Suppression Network for Weakly-supervised Temporal Action Localization' (AAAI-20 Spotlight)

background-modeling deep-learning pytorch temporal-action-localization weakly-supervised-learning

Last synced: 02 Aug 2024

https://github.com/kangyeolk/Paint-by-Sketch

Stable Diffusion-based image manipulation method with a sketch and reference image

computer-vision diffusion-models image-editing image-generation image-manipulation pytorch stable-diffusion

Last synced: 31 Oct 2024

https://github.com/kennardwang/vtuber-momosehiyori

😀 Live2D VTuber, made by Unity and Cubism. This project is based on TCP socket, OpenCV and deep learning

csharp cubism deep-learning dlib live2d opencv pycharm python3 pytorch tcp-socket unity vtuber

Last synced: 09 Oct 2024

https://github.com/matajoh/fourier_feature_nets

Supplemental learning materials for "Fourier Feature Networks and Neural Volume Rendering"

computer-graphics computer-vision deep-learning pytorch

Last synced: 31 Oct 2024

https://github.com/ai-trainee/traffic-sign-recognition-pyqt5-yolov5-gui

This YOLOv5🚀😊 GUI road sign system uses MySQL💽, PyQt5🎨, PyTorch, CSS🌈. It has modules for login🔑, YOLOv5 setup📋, sign recognition🔍, database💾, and image processing🖼️. It supports diverse inputs, model switching, and enhancements like mosaic and mixup📈.

css mysql-database pyqt5 pytorch tensorrt-engine yolov5 yolov5-gui

Last synced: 12 Oct 2024

https://github.com/ericlbuehler/xlora

X-LoRA: Mixture of LoRA Experts

llm lora python pytorch

Last synced: 30 Oct 2024

https://github.com/Js-Mim/mss_pytorch

Singing Voice Separation via Recurrent Inference and Skip-Filtering Connections - PyTorch Implementation. Demo:

deep-learning mss-pytorch music-source-separation pytorch recurrent-neural-networks skip-filtering-connections

Last synced: 03 Aug 2024

https://github.com/ahirner/pytorch-retraining

Transfer Learning Shootout for PyTorch's model zoo (torchvision)

benchmark pytorch transfer-learning

Last synced: 03 Aug 2024

https://github.com/mithril-security/bastionlab

A simple framework for privacy-friendly data science collaboration

deep-learning eda multi-party privacy pytorch secure-enclave

Last synced: 09 Nov 2024

https://github.com/chrischoy/makepytorchplusplus

How and why you want to make your pytorch CUDA/CPP extension with a Makefile

cpp-extension pytorch pytorch-cpp tutorial

Last synced: 07 Nov 2024

https://github.com/spytensor/detect_steel_bar

CCFDF AI 数钢筋大赛

detection gangjin pytorch retinanet

Last synced: 06 Nov 2024

https://github.com/IlyaGusev/summarus

Models for automatic abstractive summarization

deep-learning machine-learning nlp pytorch summarization

Last synced: 04 Nov 2024

https://github.com/daniil-osokin/gccpm-look-into-person-cvpr19.pytorch

Fast and accurate single-person pose estimation, ranked 10th at CVPR'19 LIP challenge. Contains implementation of "Global Context for Convolutional Pose Machines" paper.

convolutional-pose-machines cvpr2019 deep-learning human-pose human-pose-estimation lightweight look-into-person openvino pose-estimation pytorch real-time single-person

Last synced: 31 Oct 2024

https://github.com/ymcui/lert

LERT: A Linguistically-motivated Pre-trained Language Model(语言学信息增强的预训练模型LERT)

bert lert nlp plm pre-train pytorch tensorflow transformer

Last synced: 28 Oct 2024

https://github.com/IDSIA/modern-srwm

Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep RL Workshop) and "Accelerating Neural Self-Improvement via Bootstrapping" (ICLR 2023 Workshop)

fast-weight-programmers fast-weights pytorch reinforcement-learning rnn-based-few-shot-learning self-referential-learning self-referential-weight-matrix torchbeast transformers

Last synced: 10 Nov 2024

https://github.com/lukas-blecher/automask

Automatic Rotoscoping/Masking tool powered with Machine Learning for Blender (WIP)

blender blender-addon blender-python deep-learning machine-learning mask-segment masking motion-tracking pytorch rotoscoping

Last synced: 01 Nov 2024

https://github.com/idsia/modern-srwm

Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep RL Workshop) and "Accelerating Neural Self-Improvement via Bootstrapping" (ICLR 2023 Workshop)

fast-weight-programmers fast-weights pytorch reinforcement-learning rnn-based-few-shot-learning self-referential-learning self-referential-weight-matrix torchbeast transformers

Last synced: 11 Nov 2024

https://github.com/pxiangwu/MotionNet

CVPR 2020, "MotionNet: Joint Perception and Motion Prediction for Autonomous Driving Based on Bird's Eye View Maps"

autonomous-driving cvpr2020 motion-prediction perception pytorch

Last synced: 28 Oct 2024

https://github.com/openucx/ucc

Unified Collective Communication Library

collectives cuda deep-learning hpc infiniband mpi openshmem pgas pytorch roce sharp

Last synced: 10 Nov 2024

https://github.com/clovaai/rebias

Official Pytorch implementation of ReBias (Learning De-biased Representations with Biased Representations), ICML 2020

bias-generalisation bias-generalization deep-learning icml icml-2020 icml2020 machine-learning pytorch rebias

Last synced: 12 Nov 2024

https://github.com/prrao87/fine-grained-sentiment

A comparison and discussion of different NLP methods for 5-class sentiment classification on the SST-5 dataset.

fasttext flair nlp python pytorch sentiment-analysis text-classification transformers

Last synced: 30 Oct 2024

https://github.com/sangyx/d2l-torch

《动手学深度学习》 PyTorch 版本

chinese deep-learning jupyter-notebook python pytorch

Last synced: 31 Oct 2024

https://github.com/ducha-aiki/pytorch-sift

PyTorch implementation of SIFT descriptor

cnn descriptor image-matching local-features pytorch sift

Last synced: 01 Nov 2024

https://github.com/j3soon/tbparse

Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow

machine-learning parser plot python pytorch reader tensorboard tensorboardx tensorflow

Last synced: 03 Aug 2024

https://github.com/fastaudio/fastaudio

🔊 Audio and fastai v2

audio deep-learning fastai gpu python pytorch

Last synced: 03 Aug 2024

https://github.com/AutoVision-cloud/SA-Det3D

[ICCVW-2021] SA-Det3D: Self-attention based Context-Aware 3D Object Detection

3d-deep-learning 3d-detection autonomous-driving object-detection point-cloud pytorch

Last synced: 28 Oct 2024

https://github.com/ShomyLiu/Neu-Review-Rec

A Toolkit for Neural Review-based Recommendation models with Pytorch.

daml deepconn narre pytorch recommendation-system review-based

Last synced: 03 Aug 2024

https://github.com/rusty1s/pytorch_spline_conv

Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch

geometric-deep-learning graph-neural-networks pytorch spline-cnn

Last synced: 29 Oct 2024

https://github.com/JieShibo/PETL-ViT

[ICCV 2023] Binary Adapters, [AAAI 2023] FacT, [Tech report] Convpass

parameter-efficient-tuning pytorch transfer-learning visual-prompting

Last synced: 05 Nov 2024

https://github.com/jayleicn/TVQA

[EMNLP 2018] PyTorch code for TVQA: Localized, Compositional Video Question Answering

dataset pytorch tvqa videoqa

Last synced: 07 Nov 2024

https://github.com/nasdin/reinforcementlearning-atarigame

Pytorch LSTM RNN for reinforcement learning to play Atari games from OpenAI Universe. We also use Google Deep Mind's Asynchronous Advantage Actor-Critic (A3C) Algorithm. This is much superior and efficient than DQN and obsoletes it. Can play on many games

a3c a3c-lstm actor-critic adam asynchronous-advantage-actor-critic deep-reinforcement-learning lstm openai-gym python pytorch reinforcement-agents reinforcement-learning reinforcement-learning-algorithms rmsprop universe

Last synced: 07 Nov 2024

https://github.com/VSainteuf/pytorch-psetae

PyTorch implementation of the model presented in "Satellite Image Time Series Classification with Pixel-Set Encoders and Temporal Self-Attention"

agriculture computer-vision cvpr cvpr2020 deep-learning earth-observation machine-learning pytorch remote-sensing satellite-image self-attention spatio-temporal time-series-classification transformer-architecture

Last synced: 03 Aug 2024

https://github.com/wkcn/MobulaOP

A Simple & Flexible Cross Framework Operators Toolkit

cross-framework cupy deep-learning mxnet numpy operators pytorch scientific-computing

Last synced: 08 Nov 2024

https://github.com/xarray-contrib/xbatcher

Batch generation from xarray datasets

keras pytorch tensorflow xarray xarray-accessor

Last synced: 22 Oct 2024

https://github.com/CleanPegasus/TensorDash

TensorDash is an application that lets you remotely monitor your deep learning model's metrics and notifies you when your model training is completed or crashed.

android deep-learning fastai java keras machine-learning pypi python python-package pytorch remote-monitoring tensorflow

Last synced: 03 Aug 2024

https://github.com/Chris-hughes10/pytorch-accelerated

A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop which is flexible enough to handle the majority of use cases, and capable of utilizing different hardware options with no code changes required. Docs: https://pytorch-accelerated.readthedocs.io/en/latest/

deep-learning huggingface-accelerate huggingface-examples pytorch pytorch-tutorial timm-examples training

Last synced: 05 Aug 2024

https://github.com/shi-labs/fcf-inpainting

[WACV 2023] Keys to Better Image Inpainting: Structure and Texture Go Hand in Hand

fast-fourier-convolution fcfgan image-inpainting pytorch stylegan2

Last synced: 08 Nov 2024

https://github.com/geekjr/quickai

QuickAI is a Python library that makes it extremely easy to experiment with state-of-the-art Machine Learning models.

ai artificial-intelligence bert deep-learning dl easy-to-use fast gpt gpt-neo huggingface-transformers ml neural-network nlp object-detection python pytorch quickai research tensorflow2 yolo

Last synced: 09 Nov 2024

https://github.com/SforAiDl/vformer

A modular PyTorch library for vision transformer models

pytorch vision-transformer

Last synced: 03 Aug 2024

https://github.com/hou-yz/MVDet

[ECCV 2020] Codes and MultiviewX dataset for "Multiview Detection with Feature Perspective Transformation".

dataset detection detector multiview pedestrian-detection pytorch

Last synced: 28 Oct 2024

https://github.com/wimlds-trojmiasto/detect-waste

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

cnn deep-learning detr efficientdet fastrcnn litter maskrcnn neural-networks object-detection python pytorch trash waste-detection

Last synced: 03 Aug 2024

https://github.com/uvipen/SSD-pytorch

SSD: Single Shot MultiBox Detector pytorch implementation focusing on simplicity

computer-vision deep-learning deep-neural-networks deeplearning detection detector neural-networks object-detection pytorch ssd ssdlite

Last synced: 07 Aug 2024

https://github.com/isLinXu/YOLOv8_Efficient

🚀Simple and efficient use for Ultralytics yolov8🚀

deeplearning pytorch ultralytics yolov8

Last synced: 09 Nov 2024

https://github.com/liuzuxin/osrl

🤖 Elegant implementations of offline safe RL algorithms in PyTorch

bc-safe cdt cpq library offline-rl offline-safe-rl pytorch reinforcement-learning robotics safe-rl

Last synced: 30 Oct 2024

https://github.com/huggingface/diffusion-fast

Faster generation with text-to-image diffusion models.

diffusers diffusion-models pytorch sdxl text-to-image-generation

Last synced: 04 Nov 2024

https://github.com/nvidia/modulus-sym

Framework providing pythonic APIs, algorithms and utilities to be used with Modulus core to physics inform model training as well as higher level abstraction for domain experts

deep-learning machine-learning nvidia-gpu physics pytorch

Last synced: 29 Oct 2024

https://github.com/fcakyon/small-object-detection-benchmark

icip2022 paper: sahi benchmark on visdrone and xview datasets using fcos, vfnet and tood detectors

benchmark coco fcos icip2022 mmdetection object-detection pytorch sahi small-object-detection tood vfnet visdrone xview

Last synced: 30 Oct 2024