File size: 1,925 Bytes
c4a3298 3fc4e20 c4a3298 |
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 |
---
license: apache-2.0
---
# 🍓 Auto-Strawberry: Strawberry Daughter Plant Dataset and Pretrained Models
Welcome to the **Auto-Strawberry** project! This repository hosts the **strawberry daughter plant sample dataset** and **pretrained deep learning models** developed for automated estimation of plant growth attributes using stereo vision and deep learning techniques.
## 📦 Dataset Overview
The **Strawberry Daughter Plant Dataset** contains:
- **Multi-view RGB images:** 6 views per sample captured using a stereo vision camera setup (`cam0` and `cam1`).
- **Ground truth labels:**
- Total Leaf Area
- Fresh Mass
- Largest Petiole Length
- Average Crown Diameter
- **Format:** Images in `.jpg` and labels stored in a JSON file.
### Dataset Structure:
```plaintext
- cam0/
- cam1/
- annotations.json
```
## 🧠 Pretrained Models
### Available Backbone Models:
- **ResNet34**
- **Vision Transformer (ViT-B-16)**
- **EfficientNet B0**
### Model Weights:
Each model was trained using the **strawberry dataset** for the regression task of estimating leaf area and other growth metrics. The weights provided here can be used for inference and fine-tuning.
## 📥 Download and Usage
### Hugging Face Integration:
```bash
pip install huggingface_hub
```
```python
from huggingface_hub import hf_hub_download
# Download the model from Hugging Face Hub
hf_hub_download(repo_id="sinabjam/auto-strawberry", filename="desired_file_name.pth")
```
## 🎯 Applications
- Plant phenotyping
- Automated growth monitoring
- Precision agriculture insights
## 📜 License
This project is licensed under the **Apache 2.0 License**.
## 📬 Contact
- **Author:** Sina Baghbanijam
- **Email:** [sbaghba@ncsu.edu](mailto:sbaghba@ncsu.edu)
- **Project Repository:** [GitHub Link](https://github.com/sbaghba/Auto-Strawberry)
**🌟 If you use this dataset or models in your research, please cite this repository!** |