- Ship detection from satellite imagery using
tensorflow
andmrcnn
- Code is modified from a medium article, dataset is from kaggle ship detection challenge, sources will be added when I have time
- Set up virtual environment
cd python
conda/pipenv/virtualenv
depending on your preference
- Install requirements
pip install -r requirements.txt
- Download model and place in python directory
- Download trained model above
- Run ship infererence scripts
ship-infer-1.py
for trainingship-infer-5.py
for single file runningship-infer-8.py
for input directory running all example files in directory- run this for a demo
- output files in output directory
For problems installing tensorflow, try the following:
pip install https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-1.15.2-py3-none-any.whl
- Build docker container
- Run docker container