Top 10 Python torchvision Projects

torchinfo

3 2,280 7.1 Python

View model summaries in PyTorch!
torchgeo

2 2,223 9.7 Python

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data

Project mention: FLaNK Stack Weekly for 20 Nov 2023 | dev.to | 2023-11-20

InfluxDB

www.influxdata.com sponsored

Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
PyTorch_CIFAR10

1 599 0.0 Python

Pretrained TorchVision models on CIFAR10 dataset (with weights)
facetorch

6 403 8.7 Python

Python library for analysing faces using PyTorch
ProjectBabble

1 175 8.5 Python

An Opensource mouth tracking method for VR

Project mention: Vive focus 3 facial tracker with index | /r/ValveIndex | 2023-12-10

hm, maybe you could get the camera itself to work on PC but i don't think youd be able to switch on the LED's. If you somehow get it working, you can use it with project babble

SemanticSegmentation

1 166 0.0 Python

A framework for training segmentation models in pytorch on labelme annotations with pretrained examples of skin, cat, and pizza topping segmentation
torchSR

1 161 5.9 Python

Super Resolution datasets and models in Pytorch
WorkOS

workos.com sponsored

The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
merged_depth

3 45 1.8 Python

Monocular Depth Estimation - Weighted-average prediction from multiple pre-trained depth estimation models
pytorch-who-is-that-pokemon

1 16 4.6 Python

All 151 classes pokemon Gen1 classification with torchvision model.
HDF5ImageMarker

1 3 10.0 Python

Utility to making datasets of images and points coordinates that have been marked up on these images by user

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).

Python torchvision related posts

A wild question? Why CNNs are not aware of visual quality? [D]
2 projects | /r/deeplearning | 4 Oct 2022
Show HN: Super-resolution networks and datasets in Pytorch
1 project | news.ycombinator.com | 26 Apr 2021
Video processing for live video using resnet, processing takes longer than each frame lasts
1 project | /r/pytorch | 20 Feb 2021

Index

What are some of the best open-source torchvision projects in Python? This list will help you:

	Project	Stars
1	torchinfo	2,280
2	torchgeo	2,223
3	PyTorch_CIFAR10	599
4	facetorch	403
5	ProjectBabble	175
6	SemanticSegmentation	166
7	torchSR	161
8	merged_depth	45
9	pytorch-who-is-that-pokemon	16
10	HDF5ImageMarker	3