wdhorton / protein-atlas-fastai

Code for training a Resnet model for the Human Protein Atlas Image Classification competition

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Human Protein Atlas Image Classification Starter Pack

This is the code you need to train a resnet model and submit to the Human Protein Atlas competition. It uses the newest version (v1) of the fastai library.

To download the data, I used the Official Kaggle API package, which you can install with pip. Once installed, you can run kaggle competitions download -c human-protein-atlas-image-classification to get the data (just make sure to update the path variable in the resnet50_basic.ipynb notebook to point to the data on your machine).

Update 11/17 -- the resnet50_basic notebook doesn't work with fastai version 1.0.25 and above, so I made another notebook to work with the new data_block API. In this version, I also made changes to use the create_cnn function. You can find it at resnet50_basic_datablocks.ipynb.

About

Code for training a Resnet model for the Human Protein Atlas Image Classification competition


Languages

Language:Jupyter Notebook 99.6%Language:Python 0.4%