chengzegang

TorchSIFT

A PyTorch implementation of brute-force descriptor matcher and keypoint RANSAC, both support GPU and batch computation.

3D-Hand-Post-From-RGB-Image

Finetune Swin-Transformer to estimate 3d hand post from RGB images

PCL

A PyTorch implementation of the Prototypical Contrastive Learning pipelines (PCL) from arxiv: https://arxiv.org/abs/2005.04966

TextWorldDataset

This dataset contains sequences of states generated by playing Microsoft TextWorld. Each state contains graphs, descriptions, previous action and valid actions.

TorchTemplate

A simple Pytorch template for deep learning.

TorchVLAD

A simple implementation of differentiable VLAD in pure Pytorch (with Kornia)

aim

Aim 💫 — easy-to-use and performant open-source ML experiment tracker.

chengzegang

clearml

ClearML - Auto-Magical CI/CD to streamline your ML workflow. Experiment Manager, MLOps and Data-Management

Deep-Learning-Starter

Template repo for deep-learning

occupancy

dinov2

PyTorch code and models for the DINOv2 self-supervised learning method.

fuckslurm

A job submitter that slurm cannot stop

hatch

Modern, extensible Python project management

Insta360-Air-remap

Generate remappings for Insta360 Air spherical video with ffmpeg

ithaca365-devkit

llama

Inference code for LLaMA models

MixVPR

MixVPR: Feature Mixing for Visual Place Recognition (WACV 2023)

multi-path-transformer

my-research

nvidia-htop

A tool for enriching the output of nvidia-smi.

pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch

pytorch-NetVlad

Pytorch implementation of NetVlad including training on Pittsburgh.

rosbridge_suite

Server Implementations of the rosbridge v2 Protocol

SadTalker

[CVPR 2023] SadTalker：Learning Realistic 3D Motion Coefficients for Stylized Audio-Driven Single Image Talking Face Animation

scrap_bilibili

stablediffusion

High-Resolution Image Synthesis with Latent Diffusion Models

stella_vslam

This is a unofficial fork of OpenVSLAM (https://github.com/xdspacelab/openvslam)

TensorHive

Tool for managing exclusive GPU access for distributed machine learning workloads

TorchSLAM

A Differentiable Distributed Visual-SLAM System built on Pytorch (Currently just a piece of shit)