mispa-ms

yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

mmdeploy

OpenMMLab Model Deployment Framework

detectron2

detectron2 windows build

SlowFast

PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.

fvcore

Collection of common code that's shared among different research projects in FAIR computer vision team.

detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

yolov5_obb

yolov5 + csl_label.(Oriented Object Detection)（Rotation Detection）（Rotated BBox）基于yolov5的旋转目标检测

kinetics-dataset-downloader

Kinetics dataset downloader

ActivityNet

This repository is intended to host tools and demos for ActivityNet

kinetics-dataset

kinetics-downloader

Download DeepMind's Kinetics dataset.

pytorch-lightning

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

mmaction2

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark

pytorchvideo

A deep learning library for video understanding research.

mmtracking

OpenMMLab Video Perception Toolbox. It supports Video Object Detection (VID), Multiple Object Tracking (MOT), Single Object Tracking (SOT), Video Instance Segmentation (VIS) with a unified framework.

nnUNet

pytorch-3dunet

3D U-Net model for volumetric semantic segmentation written in pytorch

Machine-Learning-Collection

A resource for learning about Machine learning & Deep Learning

albumentations

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

elektronn3

A PyTorch-based library for working with 3D and 2D convolutional neural networks, with focus on semantic segmentation of volumetric biomedical image data

PyTorch-2D-3D-UNet-Tutorial

InnerEye-DeepLearning

Medical Imaging Deep Learning library to train and deploy 3D segmentation models on Azure Machine Learning

Image-Super-Resolution-via-Iterative-Refinement

Unofficial implementation of Image Super-Resolution via Iterative Refinement by Pytorch

amc

[ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices

CutMix-PyTorch

Official Pytorch implementation of CutMix regularizer

ConvNeXt

Code release for ConvNeXt model

Swin-Transformer-Object-Detection

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Object Detection and Instance Segmentation.

mmdetection

OpenMMLab Detection Toolbox and Benchmark

Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.

Swin-Transformer

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".