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 |
-
}</component>
|
| 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 |
|