frances-dean commited on
Commit
608ea30
1 Parent(s): 734b43b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -110,7 +110,7 @@ class Echo(torchvision.datasets.VisionDataset):
110
  self.fnames = sorted(os.listdir(self.external_test_location))
111
  else:
112
  # Load video-level labels
113
- with open(f"{self.root}FileList.csv") as f:
114
  data = pandas.read_csv(f)
115
  data["Split"].map(lambda x: x.upper())
116
 
@@ -136,7 +136,7 @@ class Echo(torchvision.datasets.VisionDataset):
136
  self.frames = collections.defaultdict(list)
137
  self.trace = collections.defaultdict(_defaultdict_of_lists)
138
 
139
- with open(f"{self.root}VolumeTracings.csv") as f:
140
  header = f.readline().strip().split(",")
141
  assert header == ["FileName", "X1", "Y1", "X2", "Y2", "Frame"]
142
 
 
110
  self.fnames = sorted(os.listdir(self.external_test_location))
111
  else:
112
  # Load video-level labels
113
+ with open(f"{self.root}/FileList.csv") as f:
114
  data = pandas.read_csv(f)
115
  data["Split"].map(lambda x: x.upper())
116
 
 
136
  self.frames = collections.defaultdict(list)
137
  self.trace = collections.defaultdict(_defaultdict_of_lists)
138
 
139
+ with open(f"{self.root}/VolumeTracings.csv") as f:
140
  header = f.readline().strip().split(",")
141
  assert header == ["FileName", "X1", "Y1", "X2", "Y2", "Frame"]
142