Transformers

Image Classification Model

Overview

This repository contains a deep learning model for image classification. The model is built using Convolutional Neural Network (CNN) architecture and trained to classify images into one of five classes: "History of Myocardinal", "Myocardinal", "Abnormal", "Normal", and "Invalid Image".

Model Details

  • Framework: TensorFlow
  • Architecture: Convolutional Neural Network (CNN)
  • Input Shape: [150, 150, 3] (RGB images)
  • Output Classes: "History of Myocardinal", "Myocardinal", "Abnormal", "Normal", "Invalid Image"

Usage

To use this model for image classification, follow these steps:

  1. Install Dependencies: Make sure you have Python installed along with the necessary libraries. You can install the required packages using the following command:
Downloads last month
8
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support