A Resnet classifier that trains on PatchCamelyon 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
#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%.