Skip to content

Training a Breat Cancer Classifier on PatchCamelyon dataset

Notifications You must be signed in to change notification settings

euwern/pcam_classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Breat Cancer Classifier on PatchCamelyon dataset

A Resnet classifier that trains on PatchCamelyon dataset

Dataset

PatchCamelyon dataset can be downloaded at https://github.com/basveeling/pcam In this repo, we assume the hdf5 files are downloaded under "/mnt/datasets/pcam/", you should have the following hdf5 files:

  • camelyonpatch_level_2_split_train_x.h5
  • camelyonpatch_level_2_split_train_y.h5
  • camelyonpatch_level_2_split_valid_x.h5
  • camelyonpatch_level_2_split_valid_y.h5
  • camelyonpatch_level_2_split_test_x.h5
  • camelyonpatch_level_2_split_test_y.h5

Training and evaluation

#running the following scripts train the model for 10 epochs and evaluate the performance on the test set. 
python train.py

We obtained a Test Accuracy of 86.32%.

About

Training a Breat Cancer Classifier on PatchCamelyon dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages