Search Packages

image-classifiers

Classification models trained on ImageNet. Keras.

flaxmodels

Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

timm-3d

Python library with Neural Networks for Volume (3D) Classification based on PyTorch.

wai-annotations

Python library for converting image and object annotations formats into each other.

neural-dream

A PyTorch implementation of DeepDream

seg-torch

Semantic Segmentation with Pytorch

pastiche

A PyTorch implementation of Neural Style Transfer (Gatys et al. 2015).

tensornets

High level network definitions with pre-trained weights in TensorFlow

wai-annotations-roi

ROI image object-detection format plugins for wai.annotations.