PyTorch | Ecosyste.ms: Awesome

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

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

Last synced: 02 Nov 2024

https://github.com/primeqa/primeqa

The prime repository for state-of-the-art Multilingual Question Answering research and development.

ai bert dpr ibm ibm-research-ai language-model machine-learning natural-language-processing neural-information-retrieval neural-search nlp python pytorch question-answering semantic-search squad transfer-learning

Last synced: 02 Jan 2025

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

https://github.com/facebookresearch/fairseq2

FAIR Sequence Modeling Toolkit 2

artificial-intelligence deep-learning machine-learning python pytorch

Last synced: 03 Jan 2025

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: 18 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/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: 12 Nov 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: 07 Jan 2025

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/modal-labs/modal-examples

Examples of programs built using Modal

cloud distributed gpu machine-learning modal python pytorch serverless stable-diffusion web

Last synced: 05 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: 03 Jan 2025

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/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: 05 Jan 2025

https://github.com/mhamilton723/STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences

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

Last synced: 15 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 Jan 2025

https://github.com/linksense/LightNet

LightNet: Light-weight Networks for Semantic Image Segmentation (Cityscapes and Mapillary Vistas Dataset)

cityscapes deeplabv3 deeplabv3plus inplace-activated-batchnorm light-weight-net light-weight-networks mapillary-vistas-dataset mobilenet mobilenet-v2 mobilenetv2 mobilenetv2plus pytorch rfmobilenetv2plus scse-aspp segmentation semantic-context-loss semantic-segmentation shufflenet wide-residual-networks

Last synced: 22 Nov 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: 12 Nov 2024

https://github.com/lucidrains/linear-attention-transformer

Transformer based on a variant of attention that is linear complexity in respect to sequence length

artificial-intelligence attention-mechanism deep-learning pytorch transformer

Last synced: 04 Jan 2025

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: 04 Jan 2025

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: 23 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/jia-zhuang/pytorch-multi-gpu-training

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

multi-gpu-training pytorch

Last synced: 03 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/mryab/efficient-dl-systems

Efficient Deep Learning Systems course materials (HSE, YSDA)

cuda deep-learning distributed-training efficient-deep-learning machine-learning ml-infrastructure mlops pytorch

Last synced: 06 Jan 2025

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: 18 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: 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: 01 Jan 2025

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

OpenMMLab FewShot Learning Toolbox and Benchmark

few-shot-learning openmmlab pytorch

Last synced: 05 Jan 2025

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 Jan 2025

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: 14 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: 03 Jan 2025

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: 27 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/cresset-template/cresset

Template repository to build PyTorch projects from source on any version of PyTorch/CUDA/cuDNN.

build cuda deep-learning deep-learning-tutorial docker docker-compose machine-learning makefile mlops mlops-template python pytorch source source-python template template-repository wheel

Last synced: 05 Nov 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: 06 Jan 2025

https://github.com/hkproj/pytorch-transformer

Attention is all you need implementation

attention-is-all-you-need paper-implementations pytorch transformer

Last synced: 05 Jan 2025

https://github.com/imoonlab/hgnn

Hypergraph Neural Networks (AAAI 2019)

complex-data deep-learning hypergraph hypergraph-neural-networks pytorch

Last synced: 04 Jan 2025

https://github.com/astrazeneca/chemicalx

A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)

biology chemistry deep-chemistry deep-learning drug drug-discovery drug-interaction drug-pair geometric-deep-learning geometry graph-neural-network machine-learning pharma polypharmacy pytorch smiles smiles-strings torch torchdrug

Last synced: 06 Jan 2025

https://github.com/AstraZeneca/chemicalx

A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)

biology chemistry deep-chemistry deep-learning drug drug-discovery drug-interaction drug-pair geometric-deep-learning geometry graph-neural-network machine-learning pharma polypharmacy pytorch smiles smiles-strings torch torchdrug

Last synced: 30 Oct 2024

https://github.com/breezedeus/cnstd

CnSTD: 基于 PyTorch/MXNet 的中文/英文场景文字检测（Scene Text Detection）、数学公式检测（Mathematical Formula Detection, MFD）、篇章分析（Layout Analysis）的Python3 包

deep-learning math-formula-detection object-detection ocr python pytorch scene-text-detection text-detection

Last synced: 02 Jan 2025

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/r9y9/nnsvs

Neural network-based singing voice synthesis library for research

deep-learning dnn kiritan python pytorch singing-synthesis singing-voice singing-voice-synthesis

Last synced: 08 Dec 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: 04 Jan 2025

https://github.com/shichenliu/condensenet

CondenseNet: Light weighted CNN for mobile devices

deep-learning mobile-device pytorch

Last synced: 04 Jan 2025

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: 18 Nov 2024

https://github.com/locuslab/qpth

A fast and differentiable QP solver for PyTorch.

deep-learning machine-learning optimization pytorch quadratic-programming

Last synced: 05 Jan 2025

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

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

numpy pytorch

Last synced: 03 Jan 2025

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: 06 Jan 2025

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/nnsvs/nnsvs

Neural network-based singing voice synthesis library for research

deep-learning dnn kiritan python pytorch singing-synthesis singing-voice singing-voice-synthesis

Last synced: 22 Nov 2024

https://github.com/zjunlp/PromptKG

PromptKG Family: a Gallery of Prompt Learning & KG-related research works, toolkits, and paper-list.

awsome-list demo-tuning dialogue genkgc knowledge-graph knowledge-informed-prompt-learning lambdakg link-prediction natural-language-processing nlp paper paper-list prompt-tuning promptkg pytorch question-answering relation-extraction retrieval-augmented retrievalre retroprompt

Last synced: 14 Nov 2024

https://github.com/limbo018/DREAMPlace

Deep learning toolkit-enabled VLSI placement

deep-learning gpu-acceleration pytorch vlsi vlsi-physical-design vlsi-placement

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

Deep learning software to decode EEG, ECG or MEG signals

deep-learning ecog eeg electrocorticography electroencephalogram electroencephalography magnetoencephalography meg neuroimaging neuroscience python pytorch

Last synced: 12 Nov 2024

https://github.com/imoonlab/deephypergraph

A pytorch library for graph and hypergraph computation.

deep-learning graph-neural-networks hypergraph-neural-networks pytorch

Last synced: 04 Jan 2025

https://github.com/NeuroDiffGym/neurodiffeq

A library for solving differential equations using neural networks based on PyTorch, used by multiple research groups around the world, including at Harvard IACS.

artificial-intelligence boundary-value-problem deep-learning differential-equations initial-value-problem mathematical-modelling neural-networks ode odes pde-solver physics-informed-neural-networks pinn pypi pytorch scientific-computing time-series

Last synced: 13 Nov 2024

https://github.com/norse/norse

Deep learning with spiking neural networks (SNNs) in PyTorch.

autograd deep-learning gpu machine-learning neural-network neuromorphic pytorch pytorch-lightning spiking-neural-networks tensor

Last synced: 02 Jan 2025

https://github.com/ahangchen/torch_base

Quickly bring up your PyTorch project(a skeleton)

computer-vision deep-learning machine-learning pytorch

Last synced: 06 Jan 2025

https://github.com/santosh-gupta/speedtorch

Library for faster pinned CPU <-> GPU transfer in Pytorch

cpu-gpu-transfer cpu-pinned-tensors cuda cuda-tensors cuda-variables cupy data-transfer embeddings embeddings-trained gpu gpu-transfer machine-learning natural-language-processing nlp pinned-cpu-tensors pytorch pytorch-tensors pytorch-variables sparse sparse-modeling

Last synced: 31 Dec 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/d-li14/mobilenetv2.pytorch

72.8% MobileNetV2 1.0 model on ImageNet and a spectrum of pre-trained MobileNetV2 models

cvpr2018 deep-neural-networks imagenet mobilenetv2 pretrained-models pytorch

Last synced: 04 Jan 2025

https://github.com/volcengine/vescale

A PyTorch Native LLM Training Framework

llm-training pytorch

Last synced: 05 Jan 2025

https://github.com/Santosh-Gupta/SpeedTorch

Library for faster pinned CPU <-> GPU transfer in Pytorch

cpu-gpu-transfer cpu-pinned-tensors cuda cuda-tensors cuda-variables cupy data-transfer embeddings embeddings-trained gpu gpu-transfer machine-learning natural-language-processing nlp pinned-cpu-tensors pytorch pytorch-tensors pytorch-variables sparse sparse-modeling

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

https://github.com/qqlu/Entity

EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation

cnn computer-vision condinst deep-learning detectron2 fcos image-segmentation instance-segmentation object-detection panoptic-segmentation pretrained-models pretrained-weights pretraining pytorch segmentation semantic-segmentation

Last synced: 31 Oct 2024

https://github.com/zdevito/aten

ATen: A TENsor library for C++11

pytorch tensor torch7

Last synced: 04 Jan 2025

https://github.com/iceychris/LibreASR

:speech_balloon: An On-Premises, Streaming Speech Recognition System

asr deep-learning esp32-lyrat fastai python pytorch rnn-transducer speech-recognition

Last synced: 05 Nov 2024

https://github.com/zhangxiaosong18/FreeAnchor

FreeAnchor: Learning to Match Anchors for Visual Object Detection (NeurIPS 2019)

computer-vision freeanchor neurips-2019 object-detection one-stage pytorch

Last synced: 26 Oct 2024

https://github.com/koursaros-ai/nboost

NBoost is a scalable, search-api-boosting platform for deploying transformer models to improve the relevance of search results on different platforms (i.e. Elasticsearch)

cloud deep-learning docker elasticsearch helm kubernetes machine-learning microservices nboost nlp proxy python pytorch search-api search-engine semantic-search tensorflow

Last synced: 01 Nov 2024

https://github.com/volcengine/veScale

A PyTorch Native LLM Training Framework

llm-training pytorch

Last synced: 04 Dec 2024

https://github.com/swz30/mirnet

[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.

attention-mechanism computer-vision eccv2020 image-denoising image-enhancement image-restoration low-level-vision multi-resolution-streams pytorch super-resolution

Last synced: 01 Jan 2025

https://github.com/ayumiymk/aster.pytorch

ASTER in Pytorch

aster computer-vision ocr pytorch scene-text text-recognition text-rectification

Last synced: 04 Jan 2025

https://github.com/deepvac/deepvac

PyTorch Project Specification.

amp coreml ddp deepvac ncnn onnx python pytorch quantization tensorboard tensorrt torchscript

Last synced: 05 Jan 2025

https://github.com/daniil-osokin/lightweight-human-pose-estimation-3d-demo.pytorch

Real-time 3D multi-person pose estimation demo in PyTorch. OpenVINO backend can be used for fast inference on CPU.

3d-human-pose cmu-panoptic computer-vision deep-learning human-pose-estimation keypoint-estimation lightweight multi-person-pose-estimation openvino pytorch real-time

Last synced: 03 Jan 2025

https://github.com/jaywalnut310/glow-tts

A Generative Flow for Text-to-Speech via Monotonic Alignment Search

deep-learning pytorch speech-synthesis text-to-speech tts

Last synced: 04 Jan 2025

https://github.com/mli0603/stereo-transformer

Revisiting Stereo Depth Estimation From a Sequence-to-Sequence Perspective with Transformers. (ICCV 2021 Oral)

pytorch stereo-depth-estimation stereo-matching stereo-transformer stereo-vision

Last synced: 02 Jan 2025

https://github.com/astra-vision/MonoScene

[CVPR 2022] "MonoScene: Monocular 3D Semantic Scene Completion": 3D Semantic Occupancy Prediction from a single image

2d-to-3d computer-vision cvpr2022 cvpr22 deep-learning kitti-360 mayavi monocular nyu-depth-v2 occupancy-prediction pytorch semantic-kitti semantic-scene-completion semantic-scene-understanding single-image-reconstruction

Last synced: 28 Oct 2024

https://github.com/Mariewelt/OpenChem

OpenChem: Deep Learning toolkit for Computational Chemistry and Drug Design Research

computational-biology computational-chemistry deep-learning deep-neural-networks drug-design drug-discovery graph-convolutional-networks machine-learning predictive-modeling pytorch qsar

Last synced: 13 Nov 2024

https://github.com/MaximeVandegar/Papers-in-100-Lines-of-Code

Implementation of papers in 100 lines of code.

3d aes artificial-intelligence deep-learning diffusion-models educational gans generative-model implementation-of-research-paper inverse-rendering machine-learning meta-learning nerf neural-radiance-fields papers python pytorch reinforcement-learning research rl

Last synced: 21 Nov 2024

https://github.com/DeepVAC/deepvac

PyTorch Project Specification.

amp coreml ddp deepvac ncnn onnx python pytorch quantization tensorboard tensorrt torchscript

Last synced: 28 Oct 2024

https://github.com/lezcano/geotorch

Constrained optimization toolkit for PyTorch

constrained-optimization invertible-neural-networks low-rank manifold-optimization orthogonality positive-definite-matrices positive-semi-definite pytorch

Last synced: 05 Jan 2025

https://github.com/Daniil-Osokin/lightweight-human-pose-estimation-3d-demo.pytorch

Real-time 3D multi-person pose estimation demo in PyTorch. OpenVINO backend can be used for fast inference on CPU.

3d-human-pose cmu-panoptic computer-vision deep-learning human-pose-estimation keypoint-estimation lightweight multi-person-pose-estimation openvino pytorch real-time

Last synced: 10 Nov 2024

https://github.com/Bobo-y/flexible-yolov5

More readable and flexible yolov5 with more backbone(gcn, resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer, etc) and (cbam，dcn and so on), and tensorrt

backbone cbam dcnv2 gcn hrnet moblienet neck object-detection ptq pytorch qat resnet shufflenet sparsity swin-transformer tensorrt triton-server yolov3 yolov5

Last synced: 09 Nov 2024

https://github.com/ThibaultGROUEIX/AtlasNet

This repository contains the source codes for the paper "AtlasNet: A Papier-Mâché Approach to Learning 3D Surface Generation ". The network is able to synthesize a mesh (point cloud + connectivity) from a low-resolution point cloud, or from an image.

3d 3d-deep-learning computer-vision cvpr2018 geometry-processing pytorch

Last synced: 28 Oct 2024

https://github.com/lorenmt/mtan

The implementation of "End-to-End Multi-Task Learning with Attention" [CVPR 2019].

attention-model multi-task-learning pytorch

Last synced: 15 Nov 2024

https://github.com/TianhongDai/reinforcement-learning-algorithms

This repository contains most of pytorch implementation based classic deep reinforcement learning algorithms, including - DQN, DDQN, Dueling Network, DDPG, SAC, A2C, PPO, TRPO. (More algorithms are still in progress)

a2c actor-critic algorithm atari2600 ddpg deep-learning deep-reinforcement-learning dqn dueling-dqn flappy-bird ppo proximal-policy-optimization pytorch sac soft-actor-critic trpo trust-region-policy-optimization

Last synced: 27 Nov 2024

https://github.com/hkproj/pytorch-stable-diffusion

Stable Diffusion implemented from scratch in PyTorch

diffusion-models latent-diffusion-models paper-implementations pytorch pytorch-implementation stable-diffusion

Last synced: 05 Jan 2025

https://github.com/msurtsukov/neural-ode

Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations

jupyter-notebook neural-ode pytorch vae

Last synced: 04 Nov 2024

https://github.com/hollobit/all-about-the-gan

All About the GANs(Generative Adversarial Networks) - Summarized lists for GAN

adversarial-nets adversarial-networks arxiv cgan classification ct-denoising deep-learning detection gan generative-adversarial-network image-synthesis medical-image-synthesis medical-imaging neural-network paper pytorch reconstruction segmentation tensorflow unsupervised-learning

Last synced: 05 Jan 2025

https://github.com/leemengtw/deep-learning-resources

由淺入深的深度學習資源 Collection of deep learning materials for everyone

chinese chinese-traditional collections courses deep-learning deeplearning gan generative-adversarial-network jupyter-notebook neural-networks python pytorch tensorflow tools

Last synced: 04 Jan 2025

https://github.com/hollobit/All-About-the-GAN

All About the GANs(Generative Adversarial Networks) - Summarized lists for GAN

adversarial-nets adversarial-networks arxiv cgan classification ct-denoising deep-learning detection gan generative-adversarial-network image-synthesis medical-image-synthesis medical-imaging neural-network paper pytorch reconstruction segmentation tensorflow unsupervised-learning

Last synced: 02 Nov 2024

https://github.com/TropComplique/mtcnn-pytorch

Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks

convolutional-neural-networks face-detection pytorch

Last synced: 27 Oct 2024

https://github.com/kaituoxu/Conv-TasNet

A PyTorch implementation of Conv-TasNet described in "TasNet: Surpassing Ideal Time-Frequency Masking for Speech Separation" with Permutation Invariant Training (PIT).

audio-separation conv-tasnet permutation-invariant-training pit pytorch source-separation speech-separation tasnet

Last synced: 02 Nov 2024

https://github.com/julesbelveze/time-series-autoencoder

PyTorch Dual-Attention LSTM-Autoencoder For Multivariate Time Series

attention-mechanisms autoencoder forecasting lstm-autoencoder multivariate-timeseries pytorch time-series

Last synced: 03 Jan 2025

https://github.com/chrischoy/fcgf

Fully Convolutional Geometric Features: Fast and accurate 3D features for registration and correspondence.

3d correspondence feature-extraction fully-convolutional minkowskiengine neural-network pytorch registration sparse-tensor

Last synced: 04 Jan 2025

https://github.com/ZhendongWang6/Uformer

[CVPR 2022] Official implementation of the paper "Uformer: A General U-Shaped Transformer for Image Restoration".

deep-learning image-deblurring image-demoireing image-denoising image-deraining image-restoration pytorch

Last synced: 15 Oct 2024

https://github.com/iMoonLab/HGNN

Hypergraph Neural Networks (AAAI 2019)

complex-data deep-learning hypergraph hypergraph-neural-networks pytorch

Last synced: 28 Oct 2024

https://github.com/ymcui/macbert

Revisiting Pre-trained Models for Chinese Natural Language Processing (MacBERT)

bert language-model macbert nlp pytorch tensorflow transformers

Last synced: 04 Jan 2025

https://github.com/swz30/MIRNet

[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.

attention-mechanism computer-vision eccv2020 image-denoising image-enhancement image-restoration low-level-vision multi-resolution-streams pytorch super-resolution

Last synced: 03 Nov 2024

https://github.com/ExponentialML/Text-To-Video-Finetuning

Finetune ModelScope's Text To Video model using Diffusers 🧨

deep-learning diffusers diffusion-models modelscope pytorch stable-diffusion text-to-video text2video

Last synced: 10 Nov 2024