kamau1 commited on
Commit
09efaa0
·
verified ·
1 Parent(s): 038d6aa

update: Readme description

Browse files
Files changed (1) hide show
  1. README.md +79 -5
README.md CHANGED
@@ -1,12 +1,86 @@
1
  ---
2
- title: Fishapi
3
- emoji: 📚
4
- colorFrom: gray
5
- colorTo: yellow
6
  sdk: docker
7
  pinned: false
8
  license: apache-2.0
9
- short_description: A simple api that uses YOLO object detection to identify fis
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