Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
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.
- GitHub: https://github.com/topics/pytorch
- Wikipedia: https://en.wikipedia.org/wiki/PyTorch
- Repo: https://github.com/pytorch/pytorch
- Created by: Facebook's AI Research lab (FAIR)
- Released: September 2016
- Related Topics: python, pytorch-tutorial,
- Last updated: 2024-12-25 00:24:13 UTC
- JSON Representation
https://github.com/fangwei123456/spikingjelly
SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.
deep-learning dvs machine-learning pytorch snn spiking-neural-networks
Last synced: 24 Dec 2024
https://github.com/kumar-shridhar/PyTorch-BayesianCNN
Bayesian Convolutional Neural Network with Variational Inference based on Bayes by Backprop in PyTorch.
aleatoric-uncertainties bayes bayes-by-backprop bayesian-convnets bayesian-deep-learning bayesian-inference bayesian-network bayesian-networks bayesian-neural-networks bayesian-statistics convolutional-neural-networks image-recognition python pytorch pytorch-cnn variational-bayes variational-inference
Last synced: 30 Oct 2024
https://github.com/speedinghzl/CCNet
CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).
cityscape pytorch scene-parsing segmentation self-attention semantic-segmentation
Last synced: 31 Oct 2024
https://github.com/nod-ai/shark-studio
SHARK Studio -- Web UI for SHARK+IREE High Performance Machine Learning Distribution
amd apple-silicon deep-learning machine-learning mlir nvidia pytorch
Last synced: 19 Dec 2024
https://github.com/diffeqml/torchdyn
A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
control-theory deep-equilibrium-models deep-learning dynamical-systems implicit-models neural-differential-equations neural-network neural-ode numerical-methods pytorch
Last synced: 21 Dec 2024
https://github.com/microsoft/mup
maximal update parametrization (µP)
deep-learning machine-learning mup mutransfer python pytorch transformers
Last synced: 24 Dec 2024
https://github.com/goatmessi8/rfbnet
Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018
detection mobilenet pytorch rfbnet
Last synced: 22 Dec 2024
https://github.com/photosynthesis-team/piq
Measures and metrics for image2image tasks. PyTorch.
brisque fid gan generative-models image-metrics image-quality image-quality-assessment image-to-image iqa kid measures metrics ms-ssim mse psnr python3 pytorch ssim vif
Last synced: 19 Dec 2024
https://github.com/GOATmessi8/RFBNet
Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018
detection mobilenet pytorch rfbnet
Last synced: 06 Nov 2024
https://github.com/charliedream1/ai_quant_trade
股票AI操盘手：从学习、模拟到实盘，一站式平台。包含股票知识、策略实例、因子挖掘、传统策略、机器学习、深度学习、强化学习、图网络、高频交易、C++部署和聚宽实例代码等，可以方便学习、模拟及实盘交易
cpp jupyter-notebook keras mlflow python pytorch sklearn tensorflow trading-bot trading-platform trading-strategies
Last synced: 19 Dec 2024
https://github.com/patrick-kidger/torchtyping
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
named-tensors python-typing pytorch shape tensors typing
Last synced: 24 Dec 2024
https://github.com/yoshitomo-matsubara/torchdistill
A coding-free framework built on PyTorch for reproducible deep learning studies. 🏆25 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.
amazon-sagemaker-lab cifar10 cifar100 coco colab-notebook glue google-colab image-classification imagenet knowledge-distillation natural-language-processing nlp object-detection pascal-voc pytorch semantic-segmentation transformer
Last synced: 18 Dec 2024
https://github.com/dmmiller612/bert-extractive-summarizer
Easy to use extractive text summarization with BERT
bert coreference extractive-summarization pytorch
Last synced: 19 Dec 2024
https://github.com/ymcui/chinese-electra
Pre-trained Chinese ELECTRA（中文ELECTRA预训练模型）
bert chinese chinese-electra electra language-model nlp pre-trained-model pytorch tensorflow
Last synced: 22 Dec 2024
https://github.com/torchgan/torchgan
Research Framework for easy and efficient training of GANs based on Pytorch
computer-vision deep-learning gans generative-adversarial-networks generative-model machine-learning neural-networks python python3 pytorch
Last synced: 20 Dec 2024
https://github.com/nerfstudio-project/nerfacc
A General NeRF Acceleration Toolbox in PyTorch.
instant-ngp nerf pytorch rendering
Last synced: 19 Dec 2024
https://github.com/qfgaohao/pytorch-ssd
MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
object-detection open-images pytorch ssd
Last synced: 21 Dec 2024
https://github.com/ymcui/Chinese-ELECTRA
Pre-trained Chinese ELECTRA（中文ELECTRA预训练模型）
bert chinese chinese-electra electra language-model nlp pre-trained-model pytorch tensorflow
Last synced: 07 Nov 2024
https://github.com/vitae-transformer/vitpose
The official repo for [NeurIPS'22] "ViTPose: Simple Vision Transformer Baselines for Human Pose Estimation" and [TPAMI'23] "ViTPose++: Vision Transformer for Generic Body Pose Estimation"
deep-learning distillation mae pose-estimation pytorch self-supervised-learning vision-transformer
Last synced: 19 Dec 2024
https://github.com/pytorch/functorch
functorch is JAX-like composable function transforms for PyTorch.
Last synced: 24 Dec 2024
https://github.com/llvm/torch-mlir
The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
Last synced: 24 Dec 2024
https://github.com/DiffEqML/torchdyn
A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
control-theory deep-equilibrium-models deep-learning dynamical-systems implicit-models neural-differential-equations neural-network neural-ode numerical-methods pytorch
Last synced: 30 Oct 2024
https://github.com/JiaRenChang/PSMNet
Pyramid Stereo Matching Network (CVPR2018)
psmnet pytorch stereo-matching stereo-vision
Last synced: 27 Oct 2024
https://github.com/lxztju/pytorch_classification
利用pytorch实现图像分类的一个完整的代码，训练，预测，TTA，模型融合，模型部署，cnn提取特征，svm或者随机森林等进行分类，模型蒸馏，一个完整的代码
cnn densenet deployment flask image-classification knn knowledge-distillation label-smoothing pytorch random-forest resnet resnext svm
Last synced: 22 Dec 2024
https://github.com/jeshraghian/snntorch
Deep and online learning with spiking neural networks in Python
machine-learning neural-networks neuron-models neuroscience python pytorch snn spike spiking spiking-neural-networks
Last synced: 24 Dec 2024
https://github.com/BachiLi/redner
Differentiable rendering without approximation.
computer-graphics computer-vision differentiable-rendering monte-carlo-ray-tracing pytorch rendering tensorflow
Last synced: 30 Oct 2024
https://github.com/bermanmaxim/lovaszsoftmax
Code for the Lovász-Softmax loss (CVPR 2018)
image-segmentation loss-functions neural-networks pytorch
Last synced: 21 Dec 2024
https://github.com/we0091234/chinese_license_plate_detection_recognition
yolov5 车牌检测 车牌识别 中文车牌识别 检测 支持12种中文车牌 支持双层车牌
deep-learning plate-recognition pytorch yolov5
Last synced: 19 Dec 2024
https://github.com/microsoft/semi-supervised-learning
A Unified Semi-Supervised Learning Codebase (NeurIPS'22)
audio-classification classification computer-vision deep-learning low-resource machine-learning natural-language-processing pytorch semi-supervised-learning semisupervised-learning transformer
Last synced: 18 Dec 2024
https://github.com/BenAAndrew/Voice-Cloning-App
A Python/Pytorch app for easily synthesising human voices
deep-learning python pytorch tacotron2 text-to-speech tts voice-cloning
Last synced: 31 Oct 2024
https://github.com/TradeMaster-NTU/TradeMaster
TradeMaster is an open-source platform for quantitative trading empowered by reinforcement learning :fire: :zap: :rainbow:
finance fintech investment-strategies jupyter-notebook machine-learning python pytorch quantitative-trading reinforcement-learning stock-market trading-platform
Last synced: 08 Nov 2024
https://github.com/declare-lab/conv-emotion
This repo contains implementation of different architectures for emotion recognition in conversations.
conversational-agents conversational-ai dialogue-systems emotion-analysis emotion-recognition emotion-recognition-in-conversation lstm memory-network natural-language-processing natural-language-understanding pretrained-models pytorch sentiment-analysis
Last synced: 21 Dec 2024
https://github.com/jrieke/traingenerator
🧙 A web app to generate template code for machine learning
deep-learning machine-learning machine-learning-tutorials pytorch pytorch-ignite pytorch-lightning scikit-learn sklearn streamlit template tensorflow webapp website
Last synced: 22 Dec 2024
https://github.com/flashinfer-ai/flashinfer
FlashInfer: Kernel Library for LLM Serving
cuda flash-attention gpu jit large-large-models llm-inference pytorch
Last synced: 19 Dec 2024
https://github.com/miracleyoo/pytorch-lightning-template
An easy/swift-to-adapt PyTorch-Lighting template. 套壳模板，简单易用，稍改原来Pytorch代码，即可适配Lightning。You can translate your previous Pytorch code much easier using this template, and keep your freedom to edit all the functions as well. Big-project-friendly as well. No need to rewrite your config in hydra.
best-practices config deep-learning pytorch pytorch-lightning pytorch-lightning-template template
Last synced: 22 Dec 2024
https://github.com/manujosephv/pytorch_tabular
A standard framework for modelling Deep Learning Models for tabular data
deep-learning hacktoberfest machine-learning pytorch pytorch-lightning tabular-data
Last synced: 24 Dec 2024
https://github.com/godweiyang/nn-cuda-example
Several simple examples for popular neural network toolkits calling custom CUDA operators.
cpp cuda neural-network python pytorch tensorflow
Last synced: 22 Dec 2024
https://github.com/facebookresearch/fastmri
A large-scale dataset of both raw MRI measurements and clinical MRI images.
convolutional-neural-networks deep-learning fastmri fastmri-challenge fastmri-dataset medical-imaging mri mri-reconstruction pytorch
Last synced: 24 Dec 2024
https://github.com/arthur151/romp
Monocular, One-stage, Regression of Multiple 3D People and their 3D positions & trajectories in camera & global coordinates. ROMP[ICCV21], BEV[CVPR22], TRACE[CVPR2023]
3d-mesh-recovery bottom-up multi-person multi-person-3d-mesh-recovery pose-estimation pytorch smpl
Last synced: 20 Dec 2024
https://github.com/mit-han-lab/torchquantum
A PyTorch-based framework for Quantum Classical Simulation, Quantum Machine Learning, Quantum Neural Networks, Parameterized Quantum Circuits with support for easy deployments on real quantum computers.
deep-learning machine-learning ml-for-systems neural-network parameterized-quantum-circuit pytorch pytorch-quantum quantum quantum-computing quantum-machine-learning quantum-neural-network quantum-simulation system
Last synced: 18 Dec 2024
https://github.com/nervanasystems/ngraph
nGraph has moved to OpenVINO
caffe2 compiler deep-learning deep-neural-networks mxnet neural-network ngraph onnx paddlepaddle performance pytorch tensorflow
Last synced: 26 Sep 2024
https://github.com/NervanaSystems/ngraph
nGraph has moved to OpenVINO
caffe2 compiler deep-learning deep-neural-networks mxnet neural-network ngraph onnx paddlepaddle performance pytorch tensorflow
Last synced: 25 Oct 2024
https://github.com/explosion/spacy-transformers
🛸 Use pretrained transformers like BERT, XLNet and GPT-2 in spaCy
bert google gpt-2 huggingface language-model machine-learning natural-language-processing natural-language-understanding nlp openai pytorch pytorch-model spacy spacy-extension spacy-pipeline transfer-learning xlnet
Last synced: 24 Dec 2024
https://github.com/takuseno/d3rlpy
An offline deep reinforcement learning library
deep-learning deep-reinforcement-learning offline-rl pytorch
Last synced: 24 Dec 2024
https://github.com/OPHoperHPO/image-background-remove-tool
✂️ Automated high-quality background removal framework for an image using neural networks. ✂️
background-removal background-removal-js basnet computer-vision deeplabv3 docker fastapi machine-learning matting python pytorch remove-background salient-object-detection segmentation torch transfer-learning trimap u2net
Last synced: 25 Oct 2024
https://github.com/ophoperhpo/image-background-remove-tool
✂️ Automated high-quality background removal framework for an image using neural networks. ✂️
background-removal background-removal-js basnet computer-vision deeplabv3 docker fastapi machine-learning matting python pytorch remove-background salient-object-detection segmentation torch transfer-learning trimap u2net
Last synced: 28 Sep 2024
https://github.com/atong01/conditional-flow-matching
TorchCFM: a Conditional Flow Matching library
continuous-normalizing-flows optimal-transport pytorch
Last synced: 21 Dec 2024
https://github.com/Arthur151/ROMP
Monocular, One-stage, Regression of Multiple 3D People and their 3D positions & trajectories in camera & global coordinates. ROMP[ICCV21], BEV[CVPR22], TRACE[CVPR2023]
3d-mesh-recovery bottom-up multi-person multi-person-3d-mesh-recovery pose-estimation pytorch smpl
Last synced: 28 Oct 2024
https://github.com/microsoft/Semi-supervised-learning
A Unified Semi-Supervised Learning Codebase (NeurIPS'22)
audio-classification classification computer-vision deep-learning low-resource machine-learning natural-language-processing pytorch semi-supervised-learning semisupervised-learning transformer
Last synced: 30 Oct 2024
https://github.com/cysu/open-reid
Open source person re-identification library in python
deep-learning metric-learning person-reidentification python pytorch
Last synced: 22 Dec 2024
https://github.com/alibaba/federatedscope
An easy-to-use federated learning platform
federated-learning machine-learning pytorch
Last synced: 19 Dec 2024
https://github.com/facebookresearch/fastMRI
A large-scale dataset of both raw MRI measurements and clinical MRI images.
convolutional-neural-networks deep-learning fastmri fastmri-challenge fastmri-dataset medical-imaging mri mri-reconstruction pytorch
Last synced: 12 Nov 2024
https://github.com/pygod-team/pygod
A Python Library for Graph Outlier Detection (Anomaly Detection)
anomaly-detection deeplearning fraud-detection graph-anomaly-detection graph-neural-networks graphmining machine-learning opensource outlier-detection python pytorch security-tools toolkit
Last synced: 24 Dec 2024
https://github.com/DefTruth/CUDA-Learn-Notes
🎉 Modern CUDA Learn Notes with PyTorch: fp32/tf32, fp16/bf16, fp8/int8, flash_attn, rope, sgemm, sgemv, warp/block reduce, dot, elementwise, softmax, layernorm, rmsnorm.
block-reduce cuda cuda-programming elementwise flash-attention flash-attention-2 flash-attention-3 gemm gemv layernorm pytorch rmsnorm softmax triton warp-reduce
Last synced: 27 Oct 2024
https://github.com/godweiyang/NN-CUDA-Example
Several simple examples for popular neural network toolkits calling custom CUDA operators.
cpp cuda neural-network python pytorch tensorflow
Last synced: 28 Oct 2024
https://github.com/nianticlabs/simplerecon
[ECCV 2022] SimpleRecon: 3D Reconstruction Without 3D Convolutions
computer-vision cost-volume depth depth-estimation eccv2022 multi-view-stereo mvs pytorch scannet visualization
Last synced: 22 Dec 2024
https://github.com/pydn/comfyui-to-python-extension
A powerful tool that translates ComfyUI workflows into executable Python code.
ai-art comfyui generative-art image-generation pytorch stable-diffusion
Last synced: 20 Dec 2024
https://github.com/borealisai/advertorch
A Toolbox for Adversarial Robustness Research
adversarial-attacks adversarial-example adversarial-examples adversarial-learning adversarial-machine-learning adversarial-perturbations benchmarking machine-learning pytorch robustness security toolbox
Last synced: 20 Dec 2024
https://github.com/fastmachinelearning/hls4ml
Machine learning on FPGAs using HLS
fpga hls intel-hls keras machine-learning neural-network onnx python pytorch vivado vivado-hls
Last synced: 24 Dec 2024
https://github.com/pydn/ComfyUI-to-Python-Extension
A powerful tool that translates ComfyUI workflows into executable Python code.
ai-art comfyui generative-art image-generation pytorch stable-diffusion
Last synced: 19 Dec 2024
https://github.com/d-li14/involution
[CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator
cvpr2021 image-classification instance-segmentation involution object-detection operator pre-trained-model pytorch semantic-segmentation
Last synced: 22 Dec 2024
https://github.com/l1aoxingyu/deep-learning-project-template
A best practice for deep learning project template architecture.
chainer deep-learning keras pytorch pytorch-tutorial
Last synced: 23 Dec 2024
https://github.com/torchssl/torchssl
A PyTorch-based library for semi-supervised learning (NeurIPS'21)
codebase computer-vision deep-learning machine-learning pytorch self-supervised-learning semi-supervised-learning toolkit
Last synced: 21 Dec 2024
https://github.com/alibaba/FederatedScope
An easy-to-use federated learning platform
federated-learning machine-learning pytorch
Last synced: 16 Nov 2024
https://github.com/shaoxiongji/federated-learning
A PyTorch Implementation of Federated Learning http://doi.org/10.5281/zenodo.4321561
deep-learning federated-learning pytorch
Last synced: 19 Dec 2024
https://github.com/BorealisAI/advertorch
A Toolbox for Adversarial Robustness Research
adversarial-attacks adversarial-example adversarial-examples adversarial-learning adversarial-machine-learning adversarial-perturbations benchmarking machine-learning pytorch robustness security toolbox
Last synced: 30 Oct 2024
https://github.com/sail-sg/poolformer
PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)
image-classification mlp pooling pytorch transformer
Last synced: 22 Dec 2024
https://github.com/TorchSSL/TorchSSL
A PyTorch-based library for semi-supervised learning (NeurIPS'21)
codebase computer-vision deep-learning machine-learning pytorch self-supervised-learning semi-supervised-learning toolkit
Last synced: 12 Nov 2024
https://github.com/stared/livelossplot
Live training loss plot in Jupyter Notebook for Keras, PyTorch and others
deep-learning jupyter-notebook keras keras-visualization pytorch
Last synced: 20 Dec 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: 20 Dec 2024
https://github.com/charlesshang/dcnv2
Deformable Convolutional Networks v2 with Pytorch
cnn deformable-convolutional-networks pytorch
Last synced: 20 Dec 2024
https://github.com/jrzaurin/pytorch-widedeep
A flexible package for multimodal-deep-learning to combine tabular data with text and images using Wide and Deep models in Pytorch
deep-learning images model-hub multimodal-deep-learning python pytorch pytorch-cv pytorch-nlp pytorch-tabular-data pytorch-transformers tabular-data text
Last synced: 18 Dec 2024
https://github.com/ai-forever/ghost
A new one shot face swap approach for image and video domains
computer-vision deep-face-swap deep-learning deepfake face-swap faceswap ghost ghost-faceswap ghost-swap pytorch
Last synced: 19 Dec 2024
https://github.com/L1aoXingyu/Deep-Learning-Project-Template
A best practice for deep learning project template architecture.
chainer deep-learning keras pytorch pytorch-tutorial
Last synced: 15 Nov 2024
https://github.com/alibaba/graph-learn
An Industrial Graph Neural Network Framework
aligraph dynamic-graph dynamic-graph-service gnn gnn-framework gnn-inference graph graph-neural-networks graphlearn pytorch tensorflow training
Last synced: 18 Dec 2024
https://github.com/mit-han-lab/data-efficient-gans
[NeurIPS 2020] Differentiable Augmentation for Data-Efficient GAN Training
data-efficient gans generative-adversarial-network image-generation neurips-2020 pytorch tensorflow
Last synced: 19 Dec 2024
https://github.com/yinboc/liif
Learning Continuous Image Representation with Local Implicit Image Function, in CVPR 2021 (Oral)
implicit-neural-representation machine-learning pytorch super-resolution
Last synced: 23 Dec 2024
https://github.com/MCG-NJU/VideoMAE
[NeurIPS 2022 Spotlight] VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training
action-recognition mae masked-autoencoder neurips-2022 pytorch self-supervised-learning transformer video-analysis video-representation-learning video-transformer video-understanding vision-transformer
Last synced: 27 Oct 2024
https://github.com/CharlesShang/DCNv2
Deformable Convolutional Networks v2 with Pytorch
cnn deformable-convolutional-networks pytorch
Last synced: 27 Oct 2024
https://github.com/wisconsinaivision/yolact_edge
The first competitive instance segmentation approach that runs on small edge devices at real-time speeds.
edge-devices instance-segmentation pytorch real-time realtime yolactedge
Last synced: 22 Dec 2024
https://github.com/jayleicn/animegan
A simple PyTorch Implementation of Generative Adversarial Networks, focusing on anime face drawing.
dataset generative-adversarial-network pytorch
Last synced: 22 Dec 2024
https://github.com/jayleicn/animeGAN
A simple PyTorch Implementation of Generative Adversarial Networks, focusing on anime face drawing.
dataset generative-adversarial-network pytorch
Last synced: 12 Nov 2024
https://github.com/nvlabs/dg-net
:couple: Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral) :couple:
apex cuhk-np dg-net dukemtmc-reid image-retrieval image-search market-1501 msmt17 open-reid person-reid person-reidentification pytorch re-identification
Last synced: 22 Dec 2024
https://github.com/hiyouga/FastEdit
🩹Editing large language models within 10 seconds⚡
bloom chatbots chatgpt falcon gpt large-language-models llama llms pytorch transformers
Last synced: 09 Nov 2024
https://github.com/hiyouga/fastedit
🩹Editing large language models within 10 seconds⚡
bloom chatbots chatgpt falcon gpt large-language-models llama llms pytorch transformers
Last synced: 21 Dec 2024
https://github.com/dalinvip/cnn-lstm-bilstm-deepcnn-clstm-in-pytorch
In PyTorch Learing Neural Networks Likes CNN、BiLSTM
cnn-bilstm cnn-model gru highway-cnn lstm-model model-bilstm pytorch torchtext
Last synced: 22 Dec 2024
https://github.com/jeff-sjtu/hybrik
Official code of "HybrIK: A Hybrid Analytical-Neural Inverse Kinematics Solution for 3D Human Pose and Shape Estimation", CVPR 2021
3d-pose-estimation cvpr cvpr21 inverse-kinematics pose-estimation pytorch smpl
Last synced: 22 Dec 2024
https://github.com/enzymead/enzyme
High-performance automatic differentiation of LLVM and MLIR.
ad automatic-differentiation c clang compiler cpp deep-learning derivative differentiable-programming enzyme gradient high-performance llvm llvm-enzyme machine-learning pytorch rust scientific-computing simulation tensorflow
Last synced: 19 Dec 2024
https://github.com/NVlabs/DG-Net
:couple: Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral) :couple:
apex cuhk-np dg-net dukemtmc-reid image-retrieval image-search market-1501 msmt17 open-reid person-reid person-reidentification pytorch re-identification
Last synced: 04 Nov 2024
https://github.com/kuaikuaikim/dface
Deep learning face detection and recognition, implemented by pytorch. (pytorch实现的人脸检测和人脸识别)
deeplearning facedetection facerecognition mtcnn mtcnn-pytorch pytorch
Last synced: 21 Dec 2024
https://github.com/pypose/pypose
A library for differentiable robotics.
autonomous-robots computer-graphics control deep-learning geometric-deep-learning kalman-filter learning lie-group optimization physics planning pose-estimation pose-graph-optimization python pytorch robotics slam
Last synced: 24 Dec 2024
https://github.com/gengyanlei/fire-smoke-detect-yolov4
fire-smoke-detect-yolov4-yolov5 and fire-smoke-detection-dataset 火灾检测，烟雾检测
darknet fire-detection fire-smoke-detection-dataset pytorch smoke-detection yolov4 yolov5
Last synced: 22 Dec 2024
https://github.com/gnes-ai/gnes
GNES is Generic Neural Elastic Search, a cloud-native semantic search system based on deep neural network.
cloud-native computer-vision database deep-learning distributed-systems dnn docker-swarm elasticsearch gnes grpc machine-learning microservices neural-network nlp python pytorch search-engine semantic-search tensorflow video-processing
Last synced: 29 Sep 2024
https://github.com/SKTBrain/KoBERT
Korean BERT pre-trained cased (KoBERT)
bert korean-nlp language-model nlp pytorch transformers
Last synced: 09 Nov 2024
https://github.com/kwea123/ngp_pl
Instant-ngp in pytorch+cuda trained with pytorch-lightning (high quality with high speed, with only few lines of legible code)
3d-reconstruction cuda instant-ngp nerf novel-view-synthesis pytorch pytorch-lightning
Last synced: 22 Dec 2024
https://github.com/WisconsinAIVision/yolact_edge
The first competitive instance segmentation approach that runs on small edge devices at real-time speeds.
edge-devices instance-segmentation pytorch real-time realtime yolactedge
Last synced: 28 Oct 2024
https://github.com/zalandoresearch/pytorch-ts
PyTorch based Probabilistic Time Series forecasting framework based on GluonTS backend
deepar lstnet n-beats probabilistic pytorch time-series
Last synced: 22 Dec 2024
https://github.com/kengz/slm-lab
Modular Deep Reinforcement Learning framework in PyTorch. Companion library of the book "Foundations of Deep Reinforcement Learning".
a2c a3c benchmark deep-reinforcement-learning dqn policy-gradient ppo pytorch reinforcement-learning sac
Last synced: 22 Dec 2024
https://github.com/kengz/SLM-Lab
Modular Deep Reinforcement Learning framework in PyTorch. Companion library of the book "Foundations of Deep Reinforcement Learning".
a2c a3c benchmark deep-reinforcement-learning dqn policy-gradient ppo pytorch reinforcement-learning sac
Last synced: 02 Nov 2024