Skip to main content

A collection of computer vision pre-trained models.

1.4k
GitHub Stars
99
Curated Resources
2
Categories
8 hours ago
Last Refreshed
Other Pre-trained ModelsModel Deployment library

Use this list with your AI agent

Add the Context Awesome MCP server to Claude, Cursor, or any MCP client, then ask:

"Show me pytorch resources from cv-pretrained-model"

Installation instructions →

What's inside

Model Deployment library

  • albumentationsPyTorch

    Fast image augmentation library.

  • android-yoloTensorflow

    Real-time object detection on Android using the YOLO network, powered by TensorFlow.

  • An End-to-End Trainable Neural Network for Image-based Sequence RecognitionPyTorch

    This software implements the Convolutional Recurrent Neural Network (CRNN), a combination of CNN, RNN and CTC loss for image-based sequence recognition tasks, such as scene text recognition and OCR.

  • AttentionOCRTensorflow

    A model for real-world image text extraction.

  • AugmentorPyTorch

    Image augmentation library in Python for machine learning.

  • CNN-LSTM-CTCMXNet

    I realize three different models for text recognition, and all of them consist of CTC loss layer to realize no segmentation for text images.

Showing a sample of 99 resources. View the full list on GitHub →