Add external url for legacy pytorch package in environment.yml and fi…

…x dataset path according to readme.md
MedicineToken · Oct 17, 2023 · 599e0e9 · 599e0e9
1 parent c3f33af
commit 599e0e9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dataset.py b/dataset.py
@@ -24,8 +24,7 @@
 class ISIC2016(Dataset):
     def __init__(self, args, data_path , transform = None, transform_msk = None, mode = 'Training',prompt = 'click', plane = False):
 
-
-        df = pd.read_csv(os.path.join(data_path, 'ISBI2016_ISIC_Part3B_' + mode + '_GroundTruth.csv'), encoding='gbk')
+        df = pd.read_csv(os.path.join(data_path, 'ISBI2016_ISIC_Part1_' + mode + '_GroundTruth.csv'), encoding='gbk')
         self.name_list = df.iloc[:,1].tolist()
         self.label_list = df.iloc[:,2].tolist()
         self.data_path = data_path
@@ -91,4 +90,4 @@ def __getitem__(self, index):
             'pt':pt,
             'image_meta_dict':image_meta_dict,
         }
-
+
diff --git a/environment.yml b/environment.yml
@@ -173,6 +173,7 @@ dependencies:
   - zlib=1.2.13=h5eee18b_0
   - zstd=1.5.5=hc292b87_0
   - pip:
+    - --extra-index-url https://download.pytorch.org/whl/cu113
     - aiosignal==1.2.0
     - alembic==1.10.4
     - appdirs==1.4.4