Spaces:
Sleeping
Sleeping
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 |
|