/PowerLine

https://www.kaggle.com/c/recognizance-2/

Primary LanguagePython

Dataset - Power Lines Detection

Using Pytorch models

Look into comments in code for better understanding and how to change

Run sh setup.sh to download certain specific packages
Run python3 train.py to train
Run python3 predict.py to make predictions
Run python3 ensemle.py to train model on ensembling other models

Using Tensorflow-kears models

The code is in form of notebooks and self-explanatory

custom.ipnb - train custom model
vgg19.ipynb - train on vgg19 architecture