diff --git a/README.md b/README.md index 6dcc87e..68df13b 100644 --- a/README.md +++ b/README.md @@ -6,6 +6,8 @@ spectrogram images. The model itself is a pretrained **FasterRCNN** Model with a **ResNet50** Backbone. Only the final predictor is replaced to not predict the 91 classes present in the coco-dataset the model is trained to but the (currently) 1 category it should detect. +HINT: Trained datasets can be found at raab@polarbear:projects/efishSignalDetector/data/rise_training + ## Long-Term and major ToDos: * implement gui to correct bounding boxes * implement reinforced learning