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/layumi/person-reid-triplet-loss

Person re-ID baseline with triplet loss

cnn image-retrieval open-reid personreid pytorch triplet-loss

Last synced: 31 Oct 2024

https://github.com/htdt/hyp_metric

Hyperbolic Vision Transformers: Combining Improvements in Metric Learning | Official repository

computer-vision machine-learning pytorch

Last synced: 15 Nov 2024

https://github.com/mblondel/fenchel-young-losses

Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses

loss-functions probabilistic-classification pytorch sklearn tensorflow

Last synced: 13 Oct 2024

https://github.com/thnkim/OpenFacePytorch

PyTorch module to use OpenFace's nn4.small2.v1.t7 model

openface pytorch

Last synced: 14 Nov 2024

https://github.com/divelab/good

GOOD: A Graph Out-of-Distribution Benchmark [NeurIPS 2022 Datasets and Benchmarks]

deep-learning distribution-shift graph-neural-networks graph-ood invariant-learning out-of-distribution-generalization pytorch pytorch-geometric

Last synced: 11 Nov 2024

https://github.com/suinleelab/path_explain

A repository for explaining feature attributions and feature interactions in deep neural networks.

explainable-ai interpretable-deep-learning machine-learning pytorch tensorflow

Last synced: 11 Nov 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: 17 Nov 2024

https://github.com/belskikh/kekas

Just another DL library

deep-learning kek neural-networks pytorch

Last synced: 15 Nov 2024

https://github.com/ben-louis/deep-image-analogy-pytorch

A python implementation of Deep-Image-Analogy based on pytorch.

pytorch style-transfer

Last synced: 15 Nov 2024

https://github.com/thinklab-sjtu/eda-ai

Implementation of NeurIPS 2021 paper "On Joint Learning for Solving Placement and Routing in Chip Design" & NeurIPS 2022 paper "The Policy-gradient Placement and Generative Routing Neural Networks for Chip Design".

deep-learning pytorch

Last synced: 13 Nov 2024

https://github.com/drsleep/refinenet-pytorch

RefineNet-101 VOC in PyTorch

pytorch refinenet semantic-segmentation

Last synced: 16 Nov 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: 11 Nov 2024

https://github.com/divelab/GOOD

GOOD: A Graph Out-of-Distribution Benchmark [NeurIPS 2022 Datasets and Benchmarks]

deep-learning distribution-shift graph-neural-networks graph-ood invariant-learning out-of-distribution-generalization pytorch pytorch-geometric

Last synced: 05 Nov 2024

https://github.com/nikhilbarhate99/min-decision-transformer

Minimal implementation of Decision Transformer: Reinforcement Learning via Sequence Modeling in PyTorch for mujoco control tasks in OpenAI gym

deep-learning deep-reinforcement-learning machine-learning mujoco offline-reinforcement-learning openai-gym pytorch pytorch-transformers reinforcement-learning robotics transformer

Last synced: 13 Nov 2024

https://github.com/sony/sqvae

Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)

bayesian deep-generative-model generative-model gumbel-softmax machine-learning pytorch vae variational-autoencoder vector-quantization vq-vae

Last synced: 16 Nov 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: 14 Nov 2024

https://github.com/ChrisWaites/pyvacy

Differentially Private Optimization for PyTorch 👁🙅‍♀️

differential-privacy pytorch

Last synced: 02 Nov 2024

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/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: 15 Nov 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/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: 16 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/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/dansuh17/alexnet-pytorch

Pytorch Implementation of AlexNet

alexnet dataset paper pytorch

Last synced: 14 Nov 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: 15 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: 15 Nov 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/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/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/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/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: 14 Nov 2024

https://github.com/czczup/urst

[AAAI 2022] Towards Ultra-Resolution Neural Style Transfer via Thumbnail Instance Normalization

neural-style-transfer pytorch

Last synced: 15 Nov 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/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/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/dengbocong/text-similarity

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

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

Last synced: 15 Nov 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/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: 15 Nov 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: 15 Nov 2024

https://github.com/guanyingc/sdps-net

Learning Based Uncalibrated Photometric Stereo for Non-Lambertian Surface (CVPR 2019 Oral)

deep-learning non-lambertian photometric-stereo pytorch uncalibrated

Last synced: 15 Nov 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/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: 13 Nov 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: 14 Nov 2024

https://github.com/ericlbuehler/xlora

X-LoRA: Mixture of LoRA Experts

llm lora python pytorch

Last synced: 13 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: 12 Oct 2024

https://github.com/adobe-research/sam_inversion

[CVPR 2022] GAN inversion and editing with spatially-adaptive multiple latent layers

computer-graphics computer-vision deep-learning gan generative-adversarial-network image-manipulation machine-learning pytorch

Last synced: 14 Nov 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/plantnet/PlantNet-300K

[NeurIPS2021] A plant image dataset with high label ambiguity and a long-tailed distribution

dataset deep-learning plants pytorch

Last synced: 15 Nov 2024

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

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

benchmark pytorch transfer-learning

Last synced: 15 Nov 2024

https://github.com/cure-lab/deciwatch

[ECCV 2022] Official implementation of the paper "DeciWatch: A Simple Baseline for 10x Efficient 2D and 3D Pose Estimation"

2d-human-pose 3d-body-recovery 3d-pose-estimation body-reconstruction deep-learning eccv eccv2022 efficiency efficient-inference efficient-neural-networks human-pose-estimation pose-estimation pytorch

Last synced: 14 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: 16 Nov 2024

https://github.com/nadeemlab/DeepLIIF

Deep Learning Inferred Multiplex ImmunoFluorescence for IHC Image Quantification (https://deepliif.org) [Nature Machine Intelligence'22, CVPR'22, MICCAI'23, Histopathology'23, MICCAI'24]

ai-ready cell-biology cell-segmentation cloud-native-applications datasets deep-learning digital-pathology image-segmentation immunohistochemical-images immunohistochemistry multiplex multitask-learning pathology pathology-gan pathology-image pytorch segmentation stitching super-resolution whole-slide-imaging

Last synced: 13 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/spytensor/detect_steel_bar

CCFDF AI 数钢筋大赛

detection gangjin pytorch retinanet

Last synced: 06 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/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: 12 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/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/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/plantnet/plantnet-300k

[NeurIPS2021] A plant image dataset with high label ambiguity and a long-tailed distribution

dataset deep-learning plants pytorch

Last synced: 14 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/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