tianhaofu Goto Github PK

py_img_seg_eval

Evaluation metrics for image segmentation inspired by paper Fully Convolutional Networks for Semantic Segmentation

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

pytorch-keypoint-detection

Pytorch FPN+ResNet+MaskRCNN Keypoint Detection

pytorch-memonger

Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174

pytorch-opcounter

Count the MACs / FLOPs of your PyTorch model.

pytorch-parallel

Optimize an example model with Python, CPP, and CUDA extensions and Ring-Allreduce.

pytorch-residual-block

A Residual Block used by ResNet - Deep Learning with Pytorch

pytorch-rnn-sequence-generation-classification

Lyrics and piano music generation in Pytorch

pytorch-roipooling

PyTorch roipooling with torchvision

pytorch-sentiment-analysis-classification

A PyTorch Tutorials of Sentiment Analysis Classification (RNN, LSTM, Bi-LSTM, LSTM+Attention, CNN)

pytorch-ssim

pytorch structural similarity (SSIM) loss

pytorch-tensor-decompositions

PyTorch implementation of [1412.6553] and [1511.06530] tensor decomposition methods for convolutional layers.

pytorch-unet

Pytorch implementation of the U-Net for image semantic segmentation, with dense CRF post-processing

pytorch_seq2seq_audio

pytorch_tricks

some tircks for PyTorch

pytorchvideo

A deep learning library for video understanding research.

qdrant

Qdrant - Vector Search Engine for the next generation of AI applications

qlib

Qlib is an AI-oriented quantitative investment platform, which aims to realize the potential, empower the research, and create the value of AI technologies in quantitative investment. With Qlib, you can easily try your ideas to create better Quant investment strategies.

quantized_neural_network

The quantization of CNN/LSTM

quantized_resnet

This repo is an implementation of quantized CNN for both weights (1-bit compression) and feature maps (2-bit compression).

quantizedlstm

Models and training scripts for "LSTMs for Keyword Spotting with ReRAM-based Compute-In-Memory Architectures" (ISCAS 2021).

racos

A theoretically-grounded derivative-free optimization method, born from a statistical view of evolutionary algorithms

raft

rangedet

Paper and Codes for “RangeDet: In Defense of Range View for LiDAR-based 3D Object Detection” (ICCV2021)

rapq

Pytorch implementation of RAPQ, IJCAI 2022

ray

An open source framework that provides a simple, universal API for building distributed applications. Ray is packaged with RLlib, a scalable reinforcement learning library, and Tune, a scalable hyperparameter tuning library.

raytracing.cuda

Ray Tracing implemented with CUDA

rbdef

rdn4depth

Region Deformer Networks for Unsupervised Depth Estimation from Unconstrained Monocular Videos, IJCAI 2019

recommendation

一些推荐系统的方法实现