머신러닝 · GitHub

MNN applications by MNN, JNI exec, RK3399. Support tflite\tensorflow\caffe\onnx models.

Code for deep generative prior (ECCV2020 oral)

Code for GAN2Shape (ICLR2021 oral)

[AAAI 2025] Official PyTorch implementation of "TinySAM: Pushing the Envelope for Efficient Segment Anything Model"

PyTorch implementation for 3D human pose estimation

Convolutional Spatial Propagation Network

Allosaurus is a pretrained universal phone recognizer for more than 2000 languages

FaceXlib aims at providing ready-to-use face-related functions based on current STOA open-source methods.

Facial-Expression-Recognition in TensorFlow. Detecting faces in video and recognize the expression(emotion).

Transformer Network for Remaining Useful Life Prediction of Lithium-Ion Batteries

基于深度学习的肿瘤辅助诊断系统，以图像分割为核心，利用人工智能完成肿瘤区域的识别勾画并提供肿瘤区域的特征来辅助医生进行诊断。有完整的模型构建、后端架设、工业级部署和前端访问功能。TensorRT、PyTorch 、OpenCV 、Flask、Vue

基于强化学习的黑神话悟空AI

ThunderGBM: Fast GBDTs and Random Forests on GPUs

A pytorch implementation of MADDPG (multi-agent deep deterministic policy gradient)

Deep learning-based outlier/anomaly detection

code and docs for the EMNLP paper "DeepPath: A Reinforcement Learning Method for Knowledge Graph Reasoning"

Visualize PyTorch tensors with a single line of code.