Commit
·
3c7ac80
1
Parent(s):
55ad83b
Update README.md
Browse files
README.md
CHANGED
|
@@ -2,3 +2,23 @@
|
|
| 2 |
license: openrail
|
| 3 |
---
|
| 4 |
# <span style="color:blue">FishNet: AI For Fish Stock Estimation</span>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
license: openrail
|
| 3 |
---
|
| 4 |
# <span style="color:blue">FishNet: AI For Fish Stock Estimation</span>
|
| 5 |
+
|
| 6 |
+
The attached model was trained on a 63000 dataset of fish images belonging to 163
|
| 7 |
+
species. First we trained a detectron2 model to detect and segment fish and feduciary markers on a board.
|
| 8 |
+
The detectron 2 model was written in pytorch, and the final classifier is a ResNet50 keras model.
|
| 9 |
+
|
| 10 |
+
Below is an example of how to use the 2 models.
|
| 11 |
+
|
| 12 |
+
## Packages
|
| 13 |
+
|
| 14 |
+
|
| 15 |
+
## Load Models
|
| 16 |
+
|
| 17 |
+
|
| 18 |
+
## load an image and transform
|
| 19 |
+
|
| 20 |
+
## run the segmentation model
|
| 21 |
+
|
| 22 |
+
##visualize
|
| 23 |
+
|
| 24 |
+
## classify the fish
|