Skip to content

tlee753/ship-detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Ship Detection

  • Ship detection from satellite imagery using tensorflow and mrcnn
  • Code is modified from a medium article, dataset is from kaggle ship detection challenge, sources will be added when I have time

Model

Python

  1. Set up virtual environment
    • cd python
    • conda/pipenv/virtualenv depending on your preference
  2. Install requirements
    • pip install -r requirements.txt
  3. Download model and place in python directory
    • Download trained model above
  4. Run ship infererence scripts
    • ship-infer-1.py for training
    • ship-infer-5.py for single file running
    • ship-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

Docker

  1. Build docker container
  2. Run docker container

Example 1

Example 2

Example 3

About

ship detection using tensorflow and mrcnn

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published