Update BreastCancerCPUv1.0.py

M-C-AD · Oct 9, 2021 · b3b17d6 · b3b17d6
1 parent 54489ba
commit b3b17d6
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/BreastCancerCPUv1.0.py b/BreastCancerCPUv1.0.py
@@ -40,9 +40,10 @@ def import_images(folder,target):
             images.append([img,target])
     return images
 
-bengin = import_images(data_dir + "/benign/",0)
-# malignant = import_images("../input/breast-ultrasound-images-dataset/Dataset_BUSI_with_GT/malignant/",1)
-# normal = import_images("../input/breast-ultrasound-images-dataset/Dataset_BUSI_with_GT/normal/",2)
+
+bengin = import_images(data_dir + "/benign",0)
+malignant = import_images(data_dir + "/malignant/",1)
+normal = import_images(data_dir + "/normal/",2)