Beckschen

Jieneng Chen's starred repositories

pytorch-lightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

Language:PythonApache-2.028109 251 7080

micrograd

A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API

Language:Jupyter NotebookMIT10136 149 30

Torch-Pruning

[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs

Language:PythonMIT2638 36 351

learn2learn

A PyTorch Library for Meta-learning Research

Language:PythonMIT2632 32 259

PyTorch-Encoding

A CV toolkit for my papers.

Language:PythonMIT2037 42 357

MedicalZooPytorch

A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation

Language:PythonMIT1712 34 25

SOTA-MedSeg

SOTA medical image segmentation methods based on various challenges

1504 76 7

Synchronized-BatchNorm-PyTorch

Synchronized Batch Normalization implementation in PyTorch.

Language:PythonMIT1501 23 43

Awesome-Edge-Detection-Papers

:books: A collection of edge/contour/boundary detection papers and toolbox.

1356 49 9

inplace_abn

In-Place Activated BatchNorm for Memory-Optimized Training of DNNs

Language:PythonBSD-3-Clause1317 39 230

fixmatch

A simple method to perform semi-supervised learning with limited data.

Language:PythonApache-2.01094 19 63

pytorch-segmentation-toolbox

PyTorch Implementations for DeeplabV3 and PSPNet

Language:PythonMIT768 15 59

Efficient-3DCNNs

PyTorch Implementation of "Resource Efficient 3D Convolutional Neural Networks", codes and pretrained models.

Language:PythonMIT767 14 42

noisystudent

Code for Noisy Student Training. https://arxiv.org/abs/1911.04252

Language:PythonApache-2.0750 29 12

MaskTextSpotterV3

The code of "Mask TextSpotter v3: Segmentation Proposal Network for Robust Scene Text Spotting"

Language:PythonNOASSERTION623 16 74

H-DenseUNet

TMI 2018. H-DenseUNet: Hybrid Densely Connected UNet for Liver and Tumor Segmentation from CT Volumes

Language:Python526 15 95

axial-deeplab

This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)

Language:PythonApache-2.0447 20 32

CCT

:page_facing_up: Semi-Supervised Semantic Segmentation with Cross-Consistency Training (CVPR 2020).

Language:PythonMIT393 10 67

EfficientNet-PyTorch-3D

A PyTorch implementation of EfficientNet

Language:PythonApache-2.0165 2 15

classification_models_3D

Set of models for classifcation of 3D volumes

Language:PythonMIT150 7 19

DBnet-lite.pytorch

A pytorch re-implementation of Real-time Scene Text Detection with Differentiable Binarization

Language:Python108 7 26

minigrad

A minimal implementation of autograd (in pure Python) 🍰

Language:Jupyter Notebook91 30

Intseg

Interactive Image Segmentation with Latent Diversity

Language:PythonMIT80 11 8

efficientnet_3D

EfficientNets in 3D variant for keras and TF.keras

Language:PythonMIT53 3 8

laploss

Laplacian Pyramid Loss

Language:PythonMIT41 1 1

HDenseUet

H-DenseUnet pytorch implementation

Language:Python21 3 5

v4r-plot

For a better scientific drawing pipeline in MATLAB 😀 or Python

Language:Jupyter Notebook15 1 2

VisualCortexCourse

Homework materials for "Probabilistic Models of the Visual Cortex"

Language:Jupyter Notebook13 50

siamese-segmentation

Image co-segmentation with a Siamese Neural Network based off of CoSegNet

Language:Jupyter Notebook11 20

Pytorch-Synchronized-BatchNorm

Synchronized batchnorm for pytorch

Language:Python9 2 3