guatavita

pytorch3d

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

clearml

ClearML - Auto-Magical CI/CD to streamline your AI workload. Experiment Management, Data Management, Pipeline, Orchestration, Scheduling & Serving in one MLOps/LLMOps solution

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

pytorch-deep-learning

Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.

nitrain

Train AI models efficiently on medical images using any framework

meshrcnn

code for Mesh R-CNN, ICCV 2019

pyntcloud

pyntcloud is a Python library for working with 3D point clouds.

ShapeFlow

Learnable Deformations Among 3D Shapes.

pyfor

Tools for analyzing aerial point clouds of forest data.

pylidar

A set of Python modules which makes it easy to write lidar processing code in Python

lidar

A Python package for delineating nested surface depressions from digital elevation data.

hdl_graph_slam

3D LIDAR-based Graph SLAM

annoy

Approximate Nearest Neighbors in C++/Python optimized for memory usage and loading/saving to disk

MITK

The Medical Imaging Interaction Toolkit.

TopologyLayer

A Topology Layer for Machine Learning : Persistent Homology + Features for PyTorch

geogram

a programming library with geometric algorithms

Spade-Normalization-Tensorflow-keras-layer

Spade Normalization as tensorflow keras layer

pixel2style2pixel

Official Implementation for "Encoding in Style: a StyleGAN Encoder for Image-to-Image Translation" (CVPR 2021) presenting the pixel2style2pixel (pSp) framework

pix2vox

Sketch-Based 3D Exploration with Stacked Generative Adversarial Networks

rt-utils

A minimal Python library to facilitate the creation and manipulation of DICOM RTStructs.

nnUNet

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

dicomrt_tools

approachingalmost

Approaching (Almost) Any Machine Learning Problem

Niffler

Niffler: A DICOM Framework for Machine Learning and Processing Pipelines.

RangerOptimizerTensorflow

This repository provides a class which can be used for the optimization of a tf.keras model. It combines the two optimizers Lookahead and RAdam to form the optimizer Ranger. According to published papers and competition results this optimizer is relatively good at training models quickly.

mayavi

3D visualization of scientific data in Python

AdaFocus

Reducing spatial redundancy in video recognition. SOTA computational efficiency.

PointCNN

PointCNN: Convolution On X-Transformed Points (NeurIPS 2018)