Embryo stage classification

Model for classifying Drosophila embryo stages

Usage:

• install conda on your laptop, if you don't have it yet
• install a new environment from the embryo_class.yml as follows:

conda env create -f embryo_class.yml

• activate the environment:

conda activate embryo_class

• use the script as follows:

python predict_from_folder.py /path/to/folder_with_images

Script options:

• --device - cpu by default, can be set to GPU number (requires a different conda env)
• --batch_size - 10 by default, reduce if script crashes/stalls
• --num_workers - 10 by default, reduce if script crashes/stalls
• --model - stage_classifier.pth by default, change when using a different model or if the model is in a different directory