MoseliMotsoehli
commited on
Commit
•
44d64be
1
Parent(s):
3d2b1c5
Update README.md
Browse files
README.md
CHANGED
@@ -4,8 +4,8 @@ license: openrail
|
|
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
|
8 |
-
The
|
9 |
|
10 |
Below is an example of how to use the 2 models.
|
11 |
|
|
|
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 fiduciary markers on a board.
|
8 |
+
The detectron2 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 |
|