Pytorch Cnn Python Packages

ptflops

Flops counter for neural networks in pytorch framework

artist-engineering-geek

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

zensols-deeplearn

General deep learning tools

fight-detection

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

trixi

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.

torch-metrics

Metrics for model evaluation in pytorch

feight-detection-from-surveillance-cameras

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

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.

timst

Python package for Image Style Transfer using pyTorch 🚀.

basicai

Set of Python classes for solving basic AI problems.

zensols-dltools

General deep learning tools

plant-disease-classification

A Python module that classifies disease on plants.