update: Readme description
Browse files
README.md
CHANGED
|
@@ -1,12 +1,86 @@
|
|
| 1 |
---
|
| 2 |
-
title:
|
| 3 |
-
emoji:
|
| 4 |
-
colorFrom:
|
| 5 |
-
colorTo:
|
| 6 |
sdk: docker
|
| 7 |
pinned: false
|
| 8 |
license: apache-2.0
|
| 9 |
-
short_description:
|
| 10 |
---
|
| 11 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
| 1 |
---
|
| 2 |
+
title: Marine Species Identification API
|
| 3 |
+
emoji: 🐟
|
| 4 |
+
colorFrom: blue
|
| 5 |
+
colorTo: indigo
|
| 6 |
sdk: docker
|
| 7 |
pinned: false
|
| 8 |
license: apache-2.0
|
| 9 |
+
short_description: marine species identification api
|
| 10 |
---
|
| 11 |
|
| 12 |
+
# Marine Species Identification API
|
| 13 |
+
|
| 14 |
+
A scalable FastAPI-based API for marine species identification using YOLOv5 deep learning model. This API can detect and classify 691+ marine species from images with high accuracy.
|
| 15 |
+
|
| 16 |
+
## Features
|
| 17 |
+
|
| 18 |
+
- **Marine Species Detection**: Identify 691+ marine species using YOLOv5
|
| 19 |
+
- **RESTful API**: Clean, well-documented REST API endpoints
|
| 20 |
+
- **Real-time Processing**: Fast inference with GPU/CPU support
|
| 21 |
+
- **Annotated Results**: Returns both detection data and annotated images
|
| 22 |
+
- **Scalable Architecture**: Built with FastAPI for high performance
|
| 23 |
+
- **HuggingFace Integration**: Seamless model loading from HuggingFace Hub
|
| 24 |
+
- **Docker Ready**: Containerized for easy deployment
|
| 25 |
+
|
| 26 |
+
## API Endpoints
|
| 27 |
+
|
| 28 |
+
### Core Endpoints
|
| 29 |
+
|
| 30 |
+
- `POST /api/v1/detect` - Detect marine species in images
|
| 31 |
+
- `GET /api/v1/species` - List all supported marine species
|
| 32 |
+
- `GET /api/v1/species/{class_id}` - Get specific species information
|
| 33 |
+
- `GET /api/v1/health` - Health check and model status
|
| 34 |
+
- `GET /api/v1/info` - API information and capabilities
|
| 35 |
+
|
| 36 |
+
### Example Usage
|
| 37 |
+
|
| 38 |
+
```bash
|
| 39 |
+
# Detect marine species
|
| 40 |
+
curl -X POST "http://localhost:7860/api/v1/detect" \
|
| 41 |
+
-H "Content-Type: application/json" \
|
| 42 |
+
-d '{
|
| 43 |
+
"image": "base64_encoded_image_data",
|
| 44 |
+
"confidence_threshold": 0.25,
|
| 45 |
+
"return_annotated_image": true
|
| 46 |
+
}'
|
| 47 |
+
```
|
| 48 |
+
|
| 49 |
+
## Development
|
| 50 |
+
|
| 51 |
+
### Local Setup
|
| 52 |
+
|
| 53 |
+
1. Clone the repository
|
| 54 |
+
2. Install dependencies: `pip install -r requirements.txt`
|
| 55 |
+
3. Run the API: `uvicorn app.main:app --host 0.0.0.0 --port 7860`
|
| 56 |
+
|
| 57 |
+
### Docker Deployment
|
| 58 |
+
|
| 59 |
+
```bash
|
| 60 |
+
docker build -t marine-species-api .
|
| 61 |
+
docker run -p 7860:7860 marine-species-api
|
| 62 |
+
```
|
| 63 |
+
|
| 64 |
+
## Documentation
|
| 65 |
+
|
| 66 |
+
- **Interactive API Docs**: `/docs` (Swagger UI)
|
| 67 |
+
- **Alternative Docs**: `/redoc` (ReDoc)
|
| 68 |
+
- **OpenAPI Schema**: `/api/v1/openapi.json`
|
| 69 |
+
|
| 70 |
+
## Model Information
|
| 71 |
+
|
| 72 |
+
- **Model**: YOLOv5 trained on marine species dataset
|
| 73 |
+
- **Classes**: 691+ marine species
|
| 74 |
+
- **Input**: RGB images (JPEG, PNG, etc.)
|
| 75 |
+
- **Output**: Bounding boxes, confidence scores, species labels
|
| 76 |
+
|
| 77 |
+
## Supported Marine Species
|
| 78 |
+
|
| 79 |
+
The model can detect a wide variety of marine life including:
|
| 80 |
+
- Fish species (Sebastes, Rockfish, etc.)
|
| 81 |
+
- Marine invertebrates (Sea stars, Urchins, etc.)
|
| 82 |
+
- Cephalopods (Octopus, Squid, etc.)
|
| 83 |
+
- Cnidarians (Jellyfish, Corals, etc.)
|
| 84 |
+
- And many more...
|
| 85 |
+
|
| 86 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|