Spaces:
Running
Running
frances-dean
commited on
Commit
•
608ea30
1
Parent(s):
734b43b
Update app.py
Browse files
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 |
|