milselarch
commited on
Commit
•
e1cca9e
1
Parent(s):
abca1b7
Upload folder using huggingface_hub
Browse files- .idea/workspace.xml +17 -13
- Loader.py +8 -4
- __pycache__/Loader.cpython-39.pyc +0 -0
- corrupted-videos.txt +0 -0
- train.py +1 -0
.idea/workspace.xml
CHANGED
@@ -5,9 +5,12 @@
|
|
5 |
</component>
|
6 |
<component name="ChangeListManager">
|
7 |
<list default="true" id="98e215cf-e237-4c1b-a854-08c5c8eeb9db" name="Changes" comment="">
|
|
|
|
|
8 |
<change beforePath="$PROJECT_DIR$/check_videos.py" beforeDir="false" afterPath="$PROJECT_DIR$/check_videos.py" afterDir="false" />
|
9 |
<change beforePath="$PROJECT_DIR$/corrupted.txt" beforeDir="false" afterPath="$PROJECT_DIR$/corrupted.txt" afterDir="false" />
|
10 |
<change beforePath="$PROJECT_DIR$/helpers.py" beforeDir="false" afterPath="$PROJECT_DIR$/helpers.py" afterDir="false" />
|
|
|
11 |
</list>
|
12 |
<option name="SHOW_DIALOG" value="false" />
|
13 |
<option name="HIGHLIGHT_CONFLICTS" value="true" />
|
@@ -39,19 +42,19 @@
|
|
39 |
<option name="hideEmptyMiddlePackages" value="true" />
|
40 |
<option name="showLibraryContents" value="true" />
|
41 |
</component>
|
42 |
-
<component name="PropertiesComponent"
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
}
|
54 |
-
}
|
55 |
<component name="RecentsManager">
|
56 |
<key name="CopyFile.RECENT_KEYS">
|
57 |
<recent name="$PROJECT_DIR$" />
|
@@ -70,7 +73,8 @@
|
|
70 |
<workItem from="1698579018554" duration="14141000" />
|
71 |
<workItem from="1698650359117" duration="2013000" />
|
72 |
<workItem from="1698671103984" duration="599000" />
|
73 |
-
<workItem from="1698775864388" duration="
|
|
|
74 |
</task>
|
75 |
<servers />
|
76 |
</component>
|
|
|
5 |
</component>
|
6 |
<component name="ChangeListManager">
|
7 |
<list default="true" id="98e215cf-e237-4c1b-a854-08c5c8eeb9db" name="Changes" comment="">
|
8 |
+
<change beforePath="$PROJECT_DIR$/.hugignore" beforeDir="false" afterPath="$PROJECT_DIR$/.hugignore" afterDir="false" />
|
9 |
+
<change beforePath="$PROJECT_DIR$/Loader.py" beforeDir="false" afterPath="$PROJECT_DIR$/Loader.py" afterDir="false" />
|
10 |
<change beforePath="$PROJECT_DIR$/check_videos.py" beforeDir="false" afterPath="$PROJECT_DIR$/check_videos.py" afterDir="false" />
|
11 |
<change beforePath="$PROJECT_DIR$/corrupted.txt" beforeDir="false" afterPath="$PROJECT_DIR$/corrupted.txt" afterDir="false" />
|
12 |
<change beforePath="$PROJECT_DIR$/helpers.py" beforeDir="false" afterPath="$PROJECT_DIR$/helpers.py" afterDir="false" />
|
13 |
+
<change beforePath="$PROJECT_DIR$/train.py" beforeDir="false" afterPath="$PROJECT_DIR$/train.py" afterDir="false" />
|
14 |
</list>
|
15 |
<option name="SHOW_DIALOG" value="false" />
|
16 |
<option name="HIGHLIGHT_CONFLICTS" value="true" />
|
|
|
42 |
<option name="hideEmptyMiddlePackages" value="true" />
|
43 |
<option name="showLibraryContents" value="true" />
|
44 |
</component>
|
45 |
+
<component name="PropertiesComponent"><![CDATA[{
|
46 |
+
"keyToString": {
|
47 |
+
"RunOnceActivity.OpenProjectViewOnStart": "true",
|
48 |
+
"RunOnceActivity.ShowReadmeOnStart": "true",
|
49 |
+
"WebServerToolWindowFactoryState": "false",
|
50 |
+
"last_opened_file_path": "/home/milselarch/projects/SUTD/capstone/dataset-downloaders",
|
51 |
+
"node.js.detected.package.eslint": "true",
|
52 |
+
"node.js.detected.package.tslint": "true",
|
53 |
+
"node.js.selected.package.eslint": "(autodetect)",
|
54 |
+
"node.js.selected.package.tslint": "(autodetect)",
|
55 |
+
"vue.rearranger.settings.migration": "true"
|
56 |
}
|
57 |
+
}]]></component>
|
58 |
<component name="RecentsManager">
|
59 |
<key name="CopyFile.RECENT_KEYS">
|
60 |
<recent name="$PROJECT_DIR$" />
|
|
|
73 |
<workItem from="1698579018554" duration="14141000" />
|
74 |
<workItem from="1698650359117" duration="2013000" />
|
75 |
<workItem from="1698671103984" duration="599000" />
|
76 |
+
<workItem from="1698775864388" duration="1530000" />
|
77 |
+
<workItem from="1698837318225" duration="2874000" />
|
78 |
</task>
|
79 |
<servers />
|
80 |
</component>
|
Loader.py
CHANGED
@@ -15,11 +15,15 @@ class GridLoader(object):
|
|
15 |
self.usable_video_filepaths = None
|
16 |
|
17 |
def load_videos(
|
18 |
-
self, verbose=False, blacklist=frozenset({
|
19 |
-
'GRID-dataset/videos/s8/lgazzs.mpg',
|
20 |
-
'GRID-dataset/videos/s8/lbwx9n.mpg'
|
21 |
-
})
|
22 |
) -> List[str]:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
23 |
usable_video_filepaths = []
|
24 |
videos_without_alignment = []
|
25 |
|
|
|
15 |
self.usable_video_filepaths = None
|
16 |
|
17 |
def load_videos(
|
18 |
+
self, verbose=False, blacklist=frozenset({})
|
|
|
|
|
|
|
19 |
) -> List[str]:
|
20 |
+
"""
|
21 |
+
:param verbose:
|
22 |
+
whether to
|
23 |
+
:param blacklist:
|
24 |
+
set of filepaths to exclude from training
|
25 |
+
:return:
|
26 |
+
"""
|
27 |
usable_video_filepaths = []
|
28 |
videos_without_alignment = []
|
29 |
|
__pycache__/Loader.cpython-39.pyc
CHANGED
Binary files a/__pycache__/Loader.cpython-39.pyc and b/__pycache__/Loader.cpython-39.pyc differ
|
|
corrupted-videos.txt
ADDED
File without changes
|
train.py
CHANGED
@@ -25,6 +25,7 @@ corrupt_video_paths = set(
|
|
25 |
|
26 |
loader = GridLoader()
|
27 |
video_filepaths = loader.load_videos(blacklist=corrupt_video_paths)
|
|
|
28 |
data = tf.data.Dataset.from_tensor_slices(video_filepaths)
|
29 |
# print('DATA', data)
|
30 |
|
|
|
25 |
|
26 |
loader = GridLoader()
|
27 |
video_filepaths = loader.load_videos(blacklist=corrupt_video_paths)
|
28 |
+
print(f'videos loaded: {len(video_filepaths)}')
|
29 |
data = tf.data.Dataset.from_tensor_slices(video_filepaths)
|
30 |
# print('DATA', data)
|
31 |
|