From 75be5f2e5ab4827c629bb0759ddd8df3a6efeece Mon Sep 17 00:00:00 2001 From: mirasdanie Date: Sat, 8 Jun 2019 15:00:28 +0300 Subject: [PATCH] Update model.py --- model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model.py b/model.py index 8bee563..19a2a8f 100644 --- a/model.py +++ b/model.py @@ -21,7 +21,7 @@ N_CLASSES = 14 CLASS_NAMES = [ 'Atelectasis', 'Cardiomegaly', 'Effusion', 'Infiltration', 'Mass', 'Nodule', 'Pneumonia', 'Pneumothorax', 'Consolidation', 'Edema', 'Emphysema', 'Fibrosis', 'Pleural_Thickening', 'Hernia'] -DATA_DIR = './ChestX-ray14/images' +DATA_DIR = '/content/gdrive/'My Drive'/images' TEST_IMAGE_LIST = './ChestX-ray14/labels/test_list.txt' BATCH_SIZE = 64 @@ -126,4 +126,4 @@ def forward(self, x): if __name__ == '__main__': - main() \ No newline at end of file + main()