triple-mu's repositories
ax-samples
Samples code for world class Artificial Intelligence SoCs for computer vision applications.
yolov5_research
add yolov7 core ,improvement research based on yolov5,SwintransformV2 and Attention Series. training skills, business customization, engineering deployment C
Ask-Anything
a simple yet interesting tool for chatting about video with chatGPT, miniGPT4 and StableLM
bitsandbytes
8-bit CUDA functions for PyTorch
detrex
detrex is a research platform for Transformer-based Object Detection algorithms including DETR (ECCV 2020), Deformable-DETR (ICLR 2021), Conditional-DETR (ICCV 2021), DAB-DETR (ICLR 2022), DN-DETR (CVPR 2022), DINO (arXiv 2022), H-DETR (arXiv 2022), etc.
DINO
Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"
EasyCV
An all-in-one toolkit for computer vision
fiftyone
The open-source tool for building high-quality datasets and computer vision models
flowvision
Datasets, Transforms and Models specific to Computer Vision
how-to-optim-algorithm-in-cuda
Cuda Learn Samples
libnpy
C++ library for reading and writing of numpy's .npy files
minicoco
Fast alternative to FiftyOne for creating a subset of the COCO dataset.
mmdetection3d
OpenMMLab's next-generation platform for general 3D object detection.
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.
nn-editor
Neural Net Editor 神经网络编辑器
onnx-modifier
A tool to modify onnx models in a visualization fashion, based on Netron and flask.
onnx-simplifier
Simplify your onnx model
onnx2torch
Convert ONNX models to PyTorch.
opencv
Open Source Computer Vision Library
OpenPCDet
OpenPCDet Toolbox for LiDAR-based 3D Object Detection.
optuna
A hyperparameter optimization framework
Painter
Painter & SegGPT Series: Vision Foundation Models from BAAI
sahi
A lightweight vision library for performing large scale object detection/ instance segmentation.
Sparsebit
A model compression and acceleration toolbox based pytorch.
stable-diffusion-webui
Stable Diffusion web UI
TransformerEngine
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper GPUs, to provide better performance with lower memory utilization in both training and inference.
yolov3
YOLOv3 in PyTorch > ONNX > CoreML > TFLite
YOLOX
YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/