/Human-Protein-Atlas-Image-Classification

Classify subcellular protein patterns in human cells

Primary LanguagePython

Human-Protein-Atlas-Image-Classification

Develop models capable of classifying mixed patterns of proteins in microscope images. (Kaggle Link)

Usage

$ python3 model.py [training data]

Packages

  • imgaug 0.2.7
  • jsonschema 2.6.0
  • pandas 0.23.4
  • Pillow 5.3.0
  • pretrainedmodels 0.7.4
  • scikit-learn 0.20.0
  • torch 0.4.1
  • torchvision 0.2.1
  • tqdm 4.29.0