🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

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

Image augmentation for machine learning experiments.

Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials.-迁移学习

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

Simple Reinforcement learning tutorials, 莫烦Python 中文AI教学

A PyTorch implementation of the Transformer model in "Attention is All You Need".

Distributed Evolutionary Algorithms in Python

Genetic Algorithm, Particle Swarm Optimization, Simulated Annealing, Ant Colony Optimization Algorithm,Immune Algorithm, Artificial Fish Swarm Algorithm, Differential Evolution and TSP(Traveling sa…

Effortless data labeling with AI support from Segment Anything and other awesome models.

京东抢购自动下单助手，GUI 支持 Windows 和 macOS

NSGA2, NSGA3, R-NSGA3, MOEAD, Genetic Algorithms (GA), Differential Evolution (DE), CMAES, PSO

Isaac Gym Reinforcement Learning Environments

Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM)

OPC UA library for python >= 3.7

Train robotic agents to learn to plan pushing and grasping actions for manipulation with deep reinforcement learning.

Reinforcement Learning Environments for Omniverse Isaac Gym

Python package for point cloud registration using probabilistic model (Coherent Point Drift, GMMReg, SVR, GMMTree, FilterReg, Bayesian CPD)

Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Lab

Python library to control a robot from 'Universal Robots' http://www.universal-robots.com/

Calibrate the camera with ZhangZhengyou method (in both distortion case and no distortion case)

Simulator of vision-based tactile sensors.

A collection of notes, codes, and pictures I have related to computer vision

Python library for controlling dynamixel motors. Documentation available here:

Tools for manipulating 3D meshes within the Menpo project.

Using Catalyst.RL to train a robot to perform peg-in-hole insertion in simulation.

This package provides a framework to automatically perform grasp tests on an arbitrary object model of choice.