Skip to main content

A Python library for predicting plant diseases from leaf images using trained deep learning models.

Project description

# ๐ŸŒฟ PlantDoc Predictor

๐ŸŒฟ PlantDoc-Predictor

PyPI version License: MIT Python Versions Downloads


A Python library for predicting plant diseases from leaf images using pre-trained or custom deep learning models.


๐Ÿš€ Features

  • โœ… Pretrained models included โ€” Ready-to-use architectures like InceptionV3, ResNet50, and MobileNetV2, trained on the 38-class PlantVillage dataset.
  • ๐Ÿง  Unified API โ€” One interface for both built-in and custom .h5 models from keras and pytorch.
  • ๐Ÿงฉ Custom model support โ€” Load your own model and label mapping JSON.
  • ๐ŸŒฑ Extensible โ€” Easily add new crops, datasets, or models via model_registry.json.
  • ๐Ÿงฐ Visualization support โ€” Displays prediction confidence and leaf images.
  • โš™๏ธ Cross-platform โ€” Works seamlessly on Windows, macOS, and Linux.

๐Ÿ”ฌ Advanced Features (NEW ๐Ÿš€) Added in release 1.0.0 and above.

  • ๐Ÿง  Full Model Access โ€” Retrieve the complete loaded model for fine-tuning and experimentation.
  • โš™๏ธ Weights Extraction โ€” Access model weights for analysis, comparison, and research.
  • ๐Ÿงฉ Layer Introspection โ€” List all layers and inspect architecture programmatically.

๐Ÿ”ฅ Feature Extraction (Research-Grade ๐Ÿš€)

  • ๐Ÿ“Š Extract intermediate representations from any layer
  • ๐Ÿงช Enables feature-space SMOTE, clustering, and embedding analysis
  • ๐Ÿง  Works across all models in the model zoo
  • ๐Ÿ” Supports custom layer selection

๐Ÿ‘‰ This transforms PlantDoc-Predictor into a feature extraction + research framework, not just an inference tool.


๐Ÿง  Why Use PlantDoc-Predictor

PlantDoc-Predictor was created to reduce repetitive work in plant-disease research.
Researchers and agritech developers often train from scratch โ€” this tool eliminates that friction by providing:

  • Pretrained baselines for benchmarking new models.
  • Standardized label sets and metadata for reproducibility.
  • Plug-and-play inference for agricultural image datasets.
  • A unified interface for rapid experimentation and deployment.

Whether youโ€™re a researcher, startup, or precision-agriculture developer, PlantDoc-Predictor simplifies your workflow and lets you focus on innovation โ€” not setup.


๐Ÿงฌ Supported Models

PlantDoc-Predictor includes a diverse model zoo of pretrained CNN architectures fine-tuned on the PlantVillage 38-class plant disease dataset in 0.2.1 version onwards.

These models are automatically downloaded from the remote registry when first used, making them ready for plug-and-play inference.

Model Name Input Size Accuracy Preprocessing Description
Recursive Additive Attention v1 224ร—224 99.7% recursive_additive_attention_cnn Custom CNN with Recursive Additive Attention (your research model)
ConvNeXt Base v1 224ร—224 99.1% convnext Modern ConvNet architecture inspired by transformers
ConvNeXt Small v1 224ร—224 99.5% convnext Lightweight ConvNeXt variant with high accuracy
ConvNeXt Tiny v1 224ร—224 99.23% convnext Efficient ConvNeXt model for faster inference
Swin Base Patch4 Window7 224ร—224 99.1% swin Hierarchical Vision Transformer with shifted windows
Swin Tiny Patch4 Window7 224ร—224 99.1% swin Lightweight Swin Transformer for efficient inference
ViT Base 16 v1 224ร—224 99.1% vit Vision Transformer base model
ViT Large 16 v1 224ร—224 99.1% vit Larger ViT model with higher capacity
ViT Small 16 v1 224ร—224 99.1% vit Smaller ViT model for faster inference
ViT Tiny 16 v1 224ร—224 99.1% vit Lightweight Vision Transformer
DenseNet169 v1 224ร—224 99.68% densenet169_v1 Best performing DenseNet variant
VGG19 v1 224ร—224 98.98% vgg19_v1 Deep VGG architecture with strong performance
DenseNet121 v1 224ร—224 98.68% densenet121_v1 Dense connectivity for efficient feature reuse
InceptionV3 v1 299ร—299 98.2% inception Inception architecture with multi-scale feature extraction
ResNet50 v1 224ร—224 97.8% resnet50 Residual network for deep feature learning
EfficientNetB50 v1 224ร—224 97.8% efficientnet Efficient scaling of CNN architecture
DenseNet210 v1 224ร—224 97.0% densenet210_v1 Very deep DenseNet variant
MobileNetV2 v1 224ร—224 96.8% mobilenetv2 Mobile-friendly lightweight architecture
VGG16 v1 224ร—224 96.8% vgg16_v1 Classic deep CNN architecture
AlexNet v1 224ร—224 92.8% alexnet_v1 Early CNN baseline model

๐Ÿงช Research Models (Published Work)

PlantDoc-Predictor also includes models derived from peer-reviewed research papers, enabling reproducibility and direct comparison with published work.

These models represent novel architectures and contributions to the field of plant disease classification.

Model Paper Authors Accuracy Description
Recursive Additive Attention v1 IEEE Paper Subham Divakar, Rojalina Priyadarshini 99.70% Custom CNN with Recursive Additive Attention mechanism for enhanced feature interaction and classification performance

๐Ÿš€ Upcoming Models

PlantDoc-Predictor is actively expanding its Model Zoo.
Future releases will include modern vision architectures and transformer-based models to further improve performance and research capabilities.

Some of the upcoming models planned for integration include:

  • Vision Transformers (ViT) โ€” e.g., vit_base_patch16_224, vit_large_patch16_224 Added in release 1.0.0 onwards
  • ConvNeXt architectures โ€” e.g., convnext_tiny, convnext_small, convnext_base Added in release 1.0.0 onwards
  • Hybrid CNNโ€“Transformer models Added in release 1.0.0 onwards
  • Swin Transformers โ€” e.g., swin_tiny_patch4_window7_224 Added in release 1.0.0 onwards
  • EfficientNetV2 family
  • Multimodal models for plant disease detection that I presented via many conferences.

Planned release very month and many more features to get added as well.

These additions will allow researchers and developers to experiment with state-of-the-art deep learning architectures for plant disease classification.

Stay tuned for future releases as the PlantDoc Model Zoo continues to grow. ๐ŸŒฟ


๐ŸŒ Community & Contact

PlantDoc-Predictor is growing into a widely used open-source machine learning library for plant disease classification.
Community participation is essential to help expand the PlantDoc Model Zoo and improve the ecosystem.

๐Ÿ“ˆ Current Usage

The library is actively used by researchers, developers, and agritech enthusiasts worldwide.

Downloads

These downloads reflect the growing interest in AI-powered plant disease detection and encourage further development of the library.


๐Ÿค Contribute to the Project

We warmly welcome contributions from the community to help improve and expand PlantDoc-Predictor.

You can contribute by:

  • ๐Ÿง  Adding new pretrained models to the PlantDoc Model Zoo
  • โš™๏ธ Improving preprocessing or inference pipelines
  • ๐Ÿ“Š Benchmarking new architectures
  • ๐Ÿงช Adding test cases and improving reproducibility
  • ๐Ÿ“š Improving documentation and tutorials
  • ๐Ÿ› Fixing bugs and optimizing performance

If you are interested in contributing to the development of this widely used ML library, feel free to open a pull request or start a discussion.


๐Ÿ“ซ Contact the Author

If you have ideas, questions, research collaboration proposals, or model suggestions, feel free to reach out.

Author: Subham Divakar
๐Ÿ“ง Email: shubham.divakar@gmail.com
๐Ÿ™ GitHub: https://github.com/shubham10divakar

We welcome researchers, ML engineers, and contributors who want to help make PlantDoc-Predictor the largest open-source model zoo for plant disease detection. ๐ŸŒฟ



๐Ÿ“Š Model Zoo Comparison

The following table compares the pretrained models included in PlantDoc-Predictor .

Model Parameters Input Size Accuracy Best Use Case
Recursive Additive Attention v1 ~12โ€“15M* 224ร—224 99.7% Research-grade model with attention (best overall)
ConvNeXt Base v1 ~89M 224ร—224 99.1% Modern CNN alternative to transformers
ConvNeXt Small v1 ~50M 224ร—224 99.5% High accuracy with better efficiency
ConvNeXt Tiny v1 ~28M 224ร—224 99.23% Efficient modern CNN
Swin Base Patch4 Window7 ~88M 224ร—224 99.1% Hierarchical Vision Transformer
Swin Tiny Patch4 Window7 ~28M 224ร—224 99.1% Lightweight transformer
ViT Large 16 v1 ~307M 224ร—224 99.1% Maximum capacity transformer
ViT Base 16 v1 ~86M 224ร—224 99.1% Standard transformer baseline
ViT Small 16 v1 ~48M 224ร—224 99.1% Balanced transformer
ViT Tiny 16 v1 ~22M 224ร—224 99.1% Lightweight transformer
DenseNet169 v1 ~14M 224ร—224 99.68% Best classical CNN
VGG19 v1 ~144M 224ร—224 98.98% High-capacity CNN
DenseNet121 v1 ~8M 224ร—224 98.68% Efficient deep CNN
InceptionV3 v1 ~23M 299ร—299 98.2% Multi-scale feature extraction
ResNet50 v1 ~25M 224ร—224 97.8% Deep residual learning
EfficientNetB50 v1 ~30M 224ร—224 97.8% Accuracy-efficiency tradeoff
DenseNet210 v1 ~20M 224ร—224 97.0% Very deep dense architecture
MobileNetV2 v1 ~3.5M 224ร—224 96.8% Mobile / edge devices
VGG16 v1 ~138M 224ร—224 96.8% Benchmark model
AlexNet v1 ~60M 224ร—224 92.8% Historical baseline

๐Ÿ† Model Performance Leaderboard

Top performing models on the PlantVillage 38-class dataset.

Rank Model Accuracy
๐Ÿฅ‡ Recursive Additive Attention v1 99.70%
๐Ÿฅˆ DenseNet169 v1 99.68%
๐Ÿฅ‰ ConvNeXt Small v1 99.50%
4 ConvNeXt Tiny v1 99.23%
5 ConvNeXt Base v1 99.10%
6 Swin Base Patch4 Window7 99.10%
7 Swin Tiny Patch4 Window7 99.10%
8 ViT Base 16 v1 99.10%
9 ViT Large 16 v1 99.10%
10 ViT Small 16 v1 99.10%
11 ViT Tiny 16 v1 99.10%
12 VGG19 v1 98.98%
13 DenseNet121 v1 98.68%
14 InceptionV3 v1 98.20%
15 ResNet50 v1 97.80%
16 EfficientNetB50 v1 97.80%
17 DenseNet210 v1 97.00%
18 MobileNetV2 v1 96.80%
19 VGG16 v1 96.80%
20 AlexNet v1 92.80%

๐Ÿ“ฆ Installation

Install directly via pip:

pip install plantdoc-predictor

๐Ÿš€ How to Use plantdoc_predictor

plantdoc_predictor provides an easy-to-use interface for plant disease prediction using multiple pretrained deep learning models trained on the PlantVillage 38-class dataset.


1๏ธโƒฃ List Available Models

You can view all available pretrained models using:

from plantdoc_predictor import predictor, Predictor

predictor.list_available_models()

Output:-

Available Models:
-----------------
- rec_add_attention_v1 | Input: [224, 224] | Acc: 99.70% | Recursive Additive Attention CNN (best overall model)
- convnext_base_v1     | Input: [224, 224] | Acc: 99.10% | ConvNeXt Base fine-tuned on PlantVillage dataset
- convnext_small_v1    | Input: [224, 224] | Acc: 99.50% | ConvNeXt Small (high accuracy + efficient)
- convnext_tiny_v1     | Input: [224, 224] | Acc: 99.23% | Lightweight ConvNeXt for fast inference
- swin_base_patch4_window7 | Input: [224, 224] | Acc: 99.10% | Swin Transformer (hierarchical vision transformer)
- swin_tiny_patch4_window7 | Input: [224, 224] | Acc: 99.10% | Lightweight Swin Transformer
- vit_base_16_v1       | Input: [224, 224] | Acc: 99.10% | Vision Transformer (base)
- vit_large_16_v1      | Input: [224, 224] | Acc: 99.10% | Vision Transformer (large)
- vit_small_16_v1      | Input: [224, 224] | Acc: 99.10% | Vision Transformer (small)
- vit_tiny_16_v1       | Input: [224, 224] | Acc: 99.10% | Vision Transformer (tiny)
- densenet169_v1       | Input: [224, 224] | Acc: 99.68% | Best-performing classical CNN
- vgg19_v1             | Input: [224, 224] | Acc: 98.98% | Deep VGG architecture
- densenet121_v1       | Input: [224, 224] | Acc: 98.68% | Efficient DenseNet variant
- inceptionv3_v1       | Input: [299, 299] | Acc: 98.20% | InceptionV3 with multi-scale feature extraction
- resnet50_v1          | Input: [224, 224] | Acc: 97.80% | Residual learning-based CNN
- efficientnetb50_v1   | Input: [224, 224] | Acc: 97.80% | EfficientNet balancing accuracy & efficiency
- densenet210_v1       | Input: [224, 224] | Acc: 97.00% | Deep DenseNet architecture
- mobilenetv2_v1       | Input: [224, 224] | Acc: 96.80% | Mobile/edge optimized model
- vgg16_v1             | Input: [224, 224] | Acc: 96.80% | Classic CNN baseline
- alexnet_v1           | Input: [224, 224] | Acc: 92.80% | Early CNN baseline

2. Choose a model from the available ones and use it below.

from plantdoc_predictor import predictor, Predictor
#predictor.list_available_models()

predictor = Predictor(model_name="efficientnetb50_v1", verbose=False)
result = predictor.predict("D:/D/my docs/my docs/projects/plant disease detection on streamlit cloud/streamlit plant disease detection/data/plantvillagedataset/train/color/Blueberry___healthy/0a3f8b2f-9bb1-4da9-85a1-fb5a52c059e2___RS_HL 2478.JPG")
print(result)

Output:-
{
 'model': 'efficientnetb50_v1',
 'label': 'Blueberry___healthy',
 'confidence': 0.9999573230743408
}

when verbose=True in Predictor(model_name="efficientnetb50_v1", verbose=True)

from plantdoc_predictor import predictor, Predictor
#predictor.list_available_models()

predictor = Predictor(model_name="efficientnetb50_v1", verbose=True)
result = predictor.predict("D:/D/my docs/my docs/projects/plant disease detection on streamlit cloud/streamlit plant disease detection/data/plantvillagedataset/train/color/Blueberry___healthy/0a3f8b2f-9bb1-4da9-85a1-fb5a52c059e2___RS_HL 2478.JPG")
print(result)

OUTPUT:-

โœ” Using cached file: efficientnetb50_v1.h5
โœ” Using cached file: efficientnetb50_v1_labels.json
WARNING:absl:Compiled the loaded model, but the compiled metrics have yet to be built. `model.compile_metrics` will be empty until you train or evaluate the model.
1/1 โ”โ”โ”โ”โ”โ”โ”โ”โ”โ”โ”โ”โ”โ”โ”โ”โ”โ”โ”โ” 2s 2s/step

================= Prediction Result =================
๐Ÿ“‚ Image Path     : D:/D/my docs/my docs/projects/plant disease detection on streamlit cloud/streamlit plant disease detection/data/plantvillagedataset/train/color/Blueberry___healthy/0a3f8b2f-9bb1-4da9-85a1-fb5a52c059e2___RS_HL 2478.JPG
๐Ÿงฉ Model Used     : efficientnetb50_v1
โœ… Predicted Class: Blueberry___healthy
๐Ÿ”ข Confidence     : 100.00%
{'model': 'efficientnetb50_v1', 'label': 'Blueberry___healthy', 'confidence': 0.9999573230743408}

๐Ÿง  Access Full Model now

You can retrieve the full Keras model for advanced use cases like fine-tuning or inspection.

from plantdoc_predictor import Predictor

predictor = Predictor(model_name="resnet50_v1")

model = predictor.get_model()

print(type(model))

Output:
%runfile 'D:/D/my docs/my docs/projects/plantdoc-predictor/plantdoc-predictor/plantdoc_predictor/Test_v1.py' --wdir
2026-03-21 12:39:04.100272: I tensorflow/core/util/port.cc:153] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable `TF_ENABLE_ONEDNN_OPTS=0`.
2026-03-21 12:39:06.296357: I tensorflow/core/util/port.cc:153] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable `TF_ENABLE_ONEDNN_OPTS=0`.
2026-03-21 12:39:06.914439: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: SSE3 SSE4.1 SSE4.2 AVX AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
WARNING:absl:Compiled the loaded model, but the compiled metrics have yet to be built. `model.compile_metrics` will be empty until you train or evaluate the model.
<class 'keras.src.models.functional.Functional'>

โš™๏ธ Extract Model Weights

Access raw model weights for research and experimentation.

from plantdoc_predictor import Predictor

predictor = Predictor(model_name="resnet50_v1")

model = predictor.get_model()
#print(type(model))
weights = predictor.get_weights()

print("Total weight tensors:", len(weights))
print("First tensor shape:", weights[0].shape)

Output:
%runfile 'D:/D/my docs/my docs/projects/plantdoc-predictor/plantdoc-predictor/plantdoc_predictor/Test_v1.py' --wdir
Reloaded modules: recursive_additive_attention_v1, predictor
WARNING:absl:Compiled the loaded model, but the compiled metrics have yet to be built. `model.compile_metrics` will be empty until you train or evaluate the model.
Total weight tensors: 320
First tensor shape: (7, 7, 3, 64)

๐Ÿงฉ Layer-wise Weight Info

Get structured information about model weights.

from plantdoc_predictor import Predictor
predictor = Predictor(model_name="resnet50_v1")

model = predictor.get_model()
#print(type(model))
weights = predictor.get_weights()

weights_info = predictor.get_weights_info()

for layer, shapes in weights_info.items():
    print(layer, shapes)

Output:
%runfile 'D:/D/my docs/my docs/projects/plantdoc-predictor/plantdoc-predictor/plantdoc_predictor/Test_v1.py' --wdir
Reloaded modules: recursive_additive_attention_v1, predictor
WARNING:absl:Compiled the loaded model, but the compiled metrics have yet to be built. `model.compile_metrics` will be empty until you train or evaluate the model.
conv1_conv [(7, 7, 3, 64), (64,)]
conv1_bn [(64,), (64,), (64,), (64,)]
conv2_block1_1_conv [(1, 1, 64, 64), (64,)]
conv2_block1_1_bn [(64,), (64,), (64,), (64,)]
conv2_block1_2_conv [(3, 3, 64, 64), (64,)]
conv2_block1_2_bn [(64,), (64,), (64,), (64,)]
conv2_block1_0_conv [(1, 1, 64, 256), (256,)]
conv2_block1_3_conv [(1, 1, 64, 256), (256,)]
conv2_block1_0_bn [(256,), (256,), (256,), (256,)]
conv2_block1_3_bn [(256,), (256,), (256,), (256,)]
conv2_block2_1_conv [(1, 1, 256, 64), (64,)]
conv2_block2_1_bn [(64,), (64,), (64,), (64,)]
conv2_block2_2_conv [(3, 3, 64, 64), (64,)]
conv2_block2_2_bn [(64,), (64,), (64,), (64,)]
conv2_block2_3_conv [(1, 1, 64, 256), (256,)]
conv2_block2_3_bn [(256,), (256,), (256,), (256,)]
conv2_block3_1_conv [(1, 1, 256, 64), (64,)]
conv2_block3_1_bn [(64,), (64,), (64,), (64,)]
conv2_block3_2_conv [(3, 3, 64, 64), (64,)]
conv2_block3_2_bn [(64,), (64,), (64,), (64,)]
conv2_block3_3_conv [(1, 1, 64, 256), (256,)]
conv2_block3_3_bn [(256,), (256,), (256,), (256,)]
conv3_block1_1_conv [(1, 1, 256, 128), (128,)]
conv3_block1_1_bn [(128,), (128,), (128,), (128,)]
conv3_block1_2_conv [(3, 3, 128, 128), (128,)]
conv3_block1_2_bn [(128,), (128,), (128,), (128,)]
conv3_block1_0_conv [(1, 1, 256, 512), (512,)]
conv3_block1_3_conv [(1, 1, 128, 512), (512,)]
conv3_block1_0_bn [(512,), (512,), (512,), (512,)]
conv3_block1_3_bn [(512,), (512,), (512,), (512,)]
conv3_block2_1_conv [(1, 1, 512, 128), (128,)]
conv3_block2_1_bn [(128,), (128,), (128,), (128,)]
conv3_block2_2_conv [(3, 3, 128, 128), (128,)]
conv3_block2_2_bn [(128,), (128,), (128,), (128,)]
conv3_block2_3_conv [(1, 1, 128, 512), (512,)]
conv3_block2_3_bn [(512,), (512,), (512,), (512,)]
conv3_block3_1_conv [(1, 1, 512, 128), (128,)]
conv3_block3_1_bn [(128,), (128,), (128,), (128,)]
conv3_block3_2_conv [(3, 3, 128, 128), (128,)]
conv3_block3_2_bn [(128,), (128,), (128,), (128,)]
conv3_block3_3_conv [(1, 1, 128, 512), (512,)]
conv3_block3_3_bn [(512,), (512,), (512,), (512,)]
conv3_block4_1_conv [(1, 1, 512, 128), (128,)]
conv3_block4_1_bn [(128,), (128,), (128,), (128,)]
conv3_block4_2_conv [(3, 3, 128, 128), (128,)]
conv3_block4_2_bn [(128,), (128,), (128,), (128,)]
conv3_block4_3_conv [(1, 1, 128, 512), (512,)]
conv3_block4_3_bn [(512,), (512,), (512,), (512,)]
conv4_block1_1_conv [(1, 1, 512, 256), (256,)]
conv4_block1_1_bn [(256,), (256,), (256,), (256,)]
conv4_block1_2_conv [(3, 3, 256, 256), (256,)]
conv4_block1_2_bn [(256,), (256,), (256,), (256,)]
conv4_block1_0_conv [(1, 1, 512, 1024), (1024,)]
conv4_block1_3_conv [(1, 1, 256, 1024), (1024,)]
conv4_block1_0_bn [(1024,), (1024,), (1024,), (1024,)]
conv4_block1_3_bn [(1024,), (1024,), (1024,), (1024,)]
conv4_block2_1_conv [(1, 1, 1024, 256), (256,)]
conv4_block2_1_bn [(256,), (256,), (256,), (256,)]
conv4_block2_2_conv [(3, 3, 256, 256), (256,)]
conv4_block2_2_bn [(256,), (256,), (256,), (256,)]
conv4_block2_3_conv [(1, 1, 256, 1024), (1024,)]
conv4_block2_3_bn [(1024,), (1024,), (1024,), (1024,)]
conv4_block3_1_conv [(1, 1, 1024, 256), (256,)]
conv4_block3_1_bn [(256,), (256,), (256,), (256,)]
conv4_block3_2_conv [(3, 3, 256, 256), (256,)]
conv4_block3_2_bn [(256,), (256,), (256,), (256,)]
conv4_block3_3_conv [(1, 1, 256, 1024), (1024,)]
conv4_block3_3_bn [(1024,), (1024,), (1024,), (1024,)]
conv4_block4_1_conv [(1, 1, 1024, 256), (256,)]
conv4_block4_1_bn [(256,), (256,), (256,), (256,)]
conv4_block4_2_conv [(3, 3, 256, 256), (256,)]
conv4_block4_2_bn [(256,), (256,), (256,), (256,)]
conv4_block4_3_conv [(1, 1, 256, 1024), (1024,)]
conv4_block4_3_bn [(1024,), (1024,), (1024,), (1024,)]
conv4_block5_1_conv [(1, 1, 1024, 256), (256,)]
conv4_block5_1_bn [(256,), (256,), (256,), (256,)]
conv4_block5_2_conv [(3, 3, 256, 256), (256,)]
conv4_block5_2_bn [(256,), (256,), (256,), (256,)]
conv4_block5_3_conv [(1, 1, 256, 1024), (1024,)]
conv4_block5_3_bn [(1024,), (1024,), (1024,), (1024,)]
conv4_block6_1_conv [(1, 1, 1024, 256), (256,)]
conv4_block6_1_bn [(256,), (256,), (256,), (256,)]
conv4_block6_2_conv [(3, 3, 256, 256), (256,)]
conv4_block6_2_bn [(256,), (256,), (256,), (256,)]
conv4_block6_3_conv [(1, 1, 256, 1024), (1024,)]
conv4_block6_3_bn [(1024,), (1024,), (1024,), (1024,)]
conv5_block1_1_conv [(1, 1, 1024, 512), (512,)]
conv5_block1_1_bn [(512,), (512,), (512,), (512,)]
conv5_block1_2_conv [(3, 3, 512, 512), (512,)]
conv5_block1_2_bn [(512,), (512,), (512,), (512,)]
conv5_block1_0_conv [(1, 1, 1024, 2048), (2048,)]
conv5_block1_3_conv [(1, 1, 512, 2048), (2048,)]
conv5_block1_0_bn [(2048,), (2048,), (2048,), (2048,)]
conv5_block1_3_bn [(2048,), (2048,), (2048,), (2048,)]
conv5_block2_1_conv [(1, 1, 2048, 512), (512,)]
conv5_block2_1_bn [(512,), (512,), (512,), (512,)]
conv5_block2_2_conv [(3, 3, 512, 512), (512,)]
conv5_block2_2_bn [(512,), (512,), (512,), (512,)]
conv5_block2_3_conv [(1, 1, 512, 2048), (2048,)]
conv5_block2_3_bn [(2048,), (2048,), (2048,), (2048,)]
conv5_block3_1_conv [(1, 1, 2048, 512), (512,)]
conv5_block3_1_bn [(512,), (512,), (512,), (512,)]
conv5_block3_2_conv [(3, 3, 512, 512), (512,)]
conv5_block3_2_bn [(512,), (512,), (512,), (512,)]
conv5_block3_3_conv [(1, 1, 512, 2048), (2048,)]
conv5_block3_3_bn [(2048,), (2048,), (2048,), (2048,)]
dense [(2048, 38), (38,)]

๐Ÿ“ License

This project is licensed under the MIT License.

You are free to:

  • โœ… Use the library for both commercial and academic purposes
  • ๐Ÿ”ง Modify, distribute, or integrate it into your own software
  • ๐ŸŒ Reference and extend it in research or production projects

Just ensure you include the original copyright notice and this license file.

See the full text in the LICENSE file.


๐ŸŒ Contributing

Contributions are warmly welcomed! ๐ŸŒฑ

We value community participation to make PlantDoc-Predictor more robust, accurate, and user-friendly.

You can contribute by:

  • ๐Ÿง  Adding new pretrained or fine-tuned models
  • โš™๏ธ Improving preprocessing or postprocessing modules
  • ๐Ÿงฉ Extending model registry or dataset support
  • ๐Ÿงช Writing new test cases for reproducibility
  • ๐Ÿ› Fixing bugs and optimizing performance
  • ๐Ÿ“š Improving documentation and adding usage examples

๐Ÿ”ง Steps to Contribute

  1. Fork this repository
  2. Create your feature branch
    git checkout -b feature/your-feature
    

๐Ÿ“ซ Contact

Author: Subham Divakar
Email: shubham.divakar@gmail.com
GitHub: shubham10divakar
PyPI: https://pypi.org/project/plantdoc-predictor/
LinkedIn: linkedin.com/in/subhamdivakar
Project Website: My Site

If you have any questions, collaboration ideas, or model suggestions โ€” feel free to reach out!
You can also open an issue or submit a pull request in the GitHub repository.


โค๏ธ Acknowledgements

A heartfelt thank you to all the open-source contributors and researchers who made this project possible:

  • ๐ŸŒฟ PlantVillage Dataset โ€” for providing an invaluable resource for agricultural disease detection.
  • ๐Ÿค– TensorFlow/Keras โ€” for powering deep learning model training and inference.
  • ๐Ÿงฉ Python Open Source Community โ€” for libraries that make AI tools easier to build.
  • ๐Ÿงช Researchers & Reviewers โ€” for advancing the field of AI in agriculture.
  • ๐Ÿ’ป Contributors โ€” for helping test, improve, and document PlantDoc-Predictor.

Your support continues to make plant disease prediction accessible and impactful across the world.


โ€œEmpowering agriculture with AI โ€” one leaf at a time.โ€ ๐ŸŒพ
โ€” Subham Divakar, Creator of PlantDoc-Predictor

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

plantdoc_predictor-1.0.2.tar.gz (32.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

plantdoc_predictor-1.0.2-py3-none-any.whl (28.8 kB view details)

Uploaded Python 3

File details

Details for the file plantdoc_predictor-1.0.2.tar.gz.

File metadata

  • Download URL: plantdoc_predictor-1.0.2.tar.gz
  • Upload date:
  • Size: 32.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.14.5

File hashes

Hashes for plantdoc_predictor-1.0.2.tar.gz
Algorithm Hash digest
SHA256 e28d6fbe43404d20ff2f9c8eb241e01c425777218769353784098fdee31b6864
MD5 89e1e9032e41c790eb32a8bf73ddb5a7
BLAKE2b-256 bc7237b2869c6685e2538149da0514c91a3965c29031ba5bb17eff2d6f802796

See more details on using hashes here.

File details

Details for the file plantdoc_predictor-1.0.2-py3-none-any.whl.

File metadata

File hashes

Hashes for plantdoc_predictor-1.0.2-py3-none-any.whl
Algorithm Hash digest
SHA256 4d885507ce2366b59770604a22fd686c7ba4cb9bace9494b2bdef7b9031b8b95
MD5 2f449ed92730df643247d21506117ee7
BLAKE2b-256 78bd719bdb15bb04fc7c186f4620d748f4a16fab2e24a655699fa50b6c56a9ed

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page