Pytorch Cnn Python Packages

ptflops

Flops counter for neural networks in pytorch framework

zensols-deeplearn

General deep learning tools

artist-engineering-geek

A bunch of GANs and data downloaders to make a custom AI artist

plant-disease-classification

A Python module that classifies disease on plants.

fight-detection

Fight Detection From Surveillance Cameras by fine-tuning a PyTorch Pretrained Model

torch-metrics

Metrics for model evaluation in pytorch

trixi-slim

Manage your machine learning experiments with trixi - modular, reproducible, high fashion. An experiment infrastructure optimized for PyTorch, but flexible enough to work for your framework and your tastes.

feight-detection-from-surveillance-cameras

Fight Detection From Surveillance Cameras by fine-tuning a PyTorch Pretrained Model

trixi

Manage your machine learning experiments with trixi - modular, reproducible, high fashion

zensols-dltools

General deep learning tools

basicai

...

timst

Python package for Image Style Transfer using pyTorch 🚀.