Skip to content

Ak1yamaKiyoshi/winstars.ai

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Implementation

ResNet34 was used as a encoder for the U-net network architecture. The network is trained on images of size 384. In the future, it can be trained to a larger size.
This approach was chosen because the ratio of pixels of ships to non-ships is very unbalanced, and it makes no sense to train the network on large images at once, as this will only increase the training time

task

Setup

  • python version: Python 3.6.4 :: Anaconda custom (64-bit)
  • install all the libraries from requirements.txt
  • Download weights and put in directory notebooks/Resnet34_lable_384_1.h5
  • Download dataset from task and put it in directory ./input/airbus-ship-detection/ <files>

Credits

It is also important to note these notebooks that helped in completing this task and understanding it.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published