raab/efishSignalDetector
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

bf

Browse Source
This commit is contained in:
Till Raab 2023-10-26 12:01:40 +02:00
parent 5c2fc32cea
commit 1258bcf1bd
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
README.md
View File

@ -52,30 +52,30 @@ im2 = ImageOps.grayscale(im1)
## config.py ## config.py
Containes Hyperparameters used by the scripts. Containes Hyperparameters used by the scripts.
## ToDos: ### ToDos:
* replace TRAIN_DIR with DATA_DIR everywhere !!! * replace TRAIN_DIR with DATA_DIR everywhere !!!
### custom_utils.py ## custom_utils.py
Classes and functions to save models and store loss values for later illustration. Classes and functions to save models and store loss values for later illustration.
Also includes helper functions... Also includes helper functions...
## ToDos: ### ToDos:
### train.py ## train.py
Code training the model using the stored images in ./data/dataset and the .csv files Code training the model using the stored images in ./data/dataset and the .csv files
containing the bounding boxes meant for training. For each epoch test-loss (without containing the bounding boxes meant for training. For each epoch test-loss (without
gradient tracking) is computed and used to infer whether the model is better than the one gradient tracking) is computed and used to infer whether the model is better than the one
of the previous epochs. If the new model is the best model, the model.state_dict is saved in of the previous epochs. If the new model is the best model, the model.state_dict is saved in
./model_outputs as best_model.pth. ./model_outputs as best_model.pth.
## ToDos: ### ToDos:
### inference.py ## inference.py
Currently, this code performs predictions based in the test dataset (img and corresponding csv file). Currently, this code performs predictions based in the test dataset (img and corresponding csv file).
However, this code shall be used to infer totally unknown images. Prediction results are ilustrated However, this code shall be used to infer totally unknown images. Prediction results are ilustrated
and stored in ./inference_output and stored in ./inference_output
## ToDo: ### ToDo:
* implement path where no csv file is needed... * implement path where no csv file is needed...