dogeplusplus/flax-u2net

Stars
3
Rank 3,945,877 (Top 79 %)
Language
Python
License
MIT License
Created over 2 years ago
Updated over 2 years ago

dogeplusplus/flax-u2net

dogeplusplus

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

JAX/Flax implementation of U-2 Net for salient object detection.

sam-at-home

Gradio UI for running Meta AI's Segment Anything on own hardware. Promptable segmentation via keypoints and bounding boxes.

cat-alan

Audio classification of domestic cats sounds (hungry, angry, purring, etc) using raw waveforms. PyTorch implementation of M5 architecture.

diffusion-mri

Diffusion model for MRI image generation

group-unet

Experiment on rotation equivariance using Group Convolution U-Nets applied to segmentation of butterfly images

ball-balancer

Reinforcement learning using PPO on the 3D Ball environment for Unity ML-Agents. Using MPI to do distributed training across separate processes

haiku-transformer

Vision Transformer implemented with JAX and dm-haiku

torch-actor-critic

PyTorch implementation of Soft Actor-Critic using MPI for distributed training.

point-network

Jaipur

Using Alpha-Beta pruning and Minimax algorithms to play the Jaipur board game with both players possessing perfect information.

termivision

Display videos and images in the terminal! (WIP)

neural-style

Web application demonstrating neural style transfer on images/webcam feeds. Built with tensorflow serving for model inference, and streamlit for UI/backend.

pensum-perdiscere