Top 7 pretrained-weight Open-Source Projects

pytorch-image-models

35 29,659 9.4 Python

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Project mention: FLaNK AI Weekly 18 March 2024 | dev.to | 2024-03-18

segmentation_models.pytorch

14 8,773 2.8 Python

Segmentation models with pretrained backbones. PyTorch.

Project mention: Instance segmentation of small objects in grainy drone imagery | /r/computervision | 2023-12-09

Also, I’d suggest considering switching to the segmentation-models library - it provides U-Net models with a variety of pretrained backbones of as encoders. The author also put out a PyTorch version. https://github.com/qubvel/segmentation_models.pytorch https://github.com/qubvel/segmentation_models

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.
super-gradients

8 4,322 9.6 Jupyter Notebook

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.

Project mention: Zero-Shot Prediction Plugin for FiftyOne | dev.to | 2024-03-13

Most computer vision models are trained to predict on a preset list of label classes. In object detection, for instance, many of the most popular models like YOLOv8 and YOLO-NAS are pretrained with the classes from the MS COCO dataset. If you download the weights checkpoints for these models and run prediction on your dataset, you will generate object detection bounding boxes for the 80 COCO classes.

Entity

2 660 6.9 Jupyter Notebook

EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation
PyTorch_CIFAR10

1 599 0.0 Python

Pretrained TorchVision models on CIFAR10 dataset (with weights)
HIPT

1 452 4.3 Jupyter Notebook

Hierarchical Image Pyramid Transformer - CVPR 2022 (Oral)
SegmentationCpp

3 400 3.4 C++

A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: VGG, ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.
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.

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).