File size: 3,797 Bytes
4f01b55 8a38f9c 4f01b55 f23ef33 0a7599e 8a38f9c 9dedc3b 8a38f9c 9dedc3b be4559a 8a38f9c 0a7599e 8a38f9c 0a7599e 8a38f9c c61c435 0a7599e 8a38f9c 0a7599e dc8870b 0a7599e 39a03b5 0a7599e dc8870b 39a03b5 0a7599e dc8870b 0a7599e dc8870b 7bfa32a 8a38f9c 443192c 8a38f9c 951889b 8a38f9c 951889b 1509372 c61c435 8a38f9c 7bfa32a 8a38f9c 951889b 8a38f9c 7f569e3 8a38f9c 0a7599e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 | ---
language: en
library_name: keras
tags:
- image-classification
- transfer-learning
- computer-vision
- keras
- tensorflow
- multiclass-classification
license: mit
datasets:
- custom
model-type: multi-model-repository
author: Thyagharajan K K
pipeline_tag: image-classification
inference: true
app_file: app.py
---
# π§ KKT-HF-TransferLearning-Models
## **Ready-to-Use Transfer Learning Models for Image Classification**
**Created by [Thyagharajan K K](https://huggingface.co/kkthyagharajan)**
[](https://huggingface.co/kkthyagharajan/KKT-HF-TransferLearning-Models)
[](https://opensource.org/licenses/MIT)
[](https://keras.io/)
[](https://www.tensorflow.org/)
[](https://github.com/kkthyagharajan/KKT_DL_Package)
---
## π Overview
This repository hosts a **collection of pretrained image-classification models** created using **Transfer Learning** in Keras/TensorFlow.
Each subdirectory contains:
- A trained `.keras` model file
- A `class_names.txt` file
- A `Test` folder with example test images
You can use these models **programmatically** or through an **interactive demo app** powered by Gradio or Streamlit.
---
## π§© Programmatic Usage
```python
from KKT_DL_Package.utils.KKT_predictions import (
multiclass_prediction_return,
display_images_gui,
get_hf_model_img_labels_local_path,
)
IMG_SIZE = (300, 300)
model_full_path, test_folder_path, class_names = (
get_hf_model_img_labels_local_path(
"kkthyagharajan/KKT-HF-TransferLearning-Models",
IMG_SIZE,
force_refresh=False, # Wonβt redownload if cached
)
)
# Run predictions and display results
all_image_paths, all_predicted_labels, all_confidences = (
multiclass_prediction_return(
model_full_path,
test_folder_path,
class_names,
IMG_SIZE,
)
)
display_images_gui(
all_image_paths,
all_predicted_labels,
IMG_SIZE,
)
```
---
## π Interactive Demo App
### π§© Option 1: Run directly on Hugging Face
This Space includes a web app defined by `app.py`.
### π» Option 2: Run locally using Gradio or Streamlit
```bash
pip install -r requirements.txt
python app.py # or
streamlit run app.py
```
---
## π Repository Structure
```
KKT-HF-TransferLearning-Models/ β Root directory (your HF repo root)
β
βββ Insect_Inception_V3/ β Model 1 directory
β βββ insect_inception_v3_model.keras β Saved model file
β βββ class_names.txt β Corresponding class labels
β βββ InsectTest/ β Test image folder
β βββ image_001.jpg
β βββ image_002.jpg
β βββ ...
β
βββ Insect_MobileNetV2/ β Model 2 directory (example)
β βββ insect_mobilenet_v2_model.keras
β βββ class_names.txt
β βββ InsectTest/
β βββ test1.jpg
β βββ test2.jpg
β βββ ...
β
βββ Insect_ResNet50/
β βββ insect_resnet50_model.keras
β βββ class_names.txt
β βββ InsectTest/
β βββ imgA.jpg
β βββ imgB.jpg
β βββ ...
β
βββ README.md
```
|