The dependencies are listed in the requirement.txt
.
Use pip3 install requirement.txt
to install them.
The major pipeline is implemented in the main.ipynb file under the notebooks folder
. A couple of trained models are accessible in the model
folder.
This work is developed based on the tutorial published by the Pytorch community https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html and this course https://github.com/akTwelve/cocosynth.
Some prediction results on the CWFID and real garden images taken on Wilson center's roof.
Artificial Dataset (images and masks):