milselarch commited on
Commit
e1cca9e
1 Parent(s): abca1b7

Upload folder using huggingface_hub

Browse files
.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
- &quot;keyToString&quot;: {
44
- &quot;RunOnceActivity.OpenProjectViewOnStart&quot;: &quot;true&quot;,
45
- &quot;RunOnceActivity.ShowReadmeOnStart&quot;: &quot;true&quot;,
46
- &quot;WebServerToolWindowFactoryState&quot;: &quot;false&quot;,
47
- &quot;last_opened_file_path&quot;: &quot;/home/milselarch/projects/SUTD/50-035/lipread-project&quot;,
48
- &quot;node.js.detected.package.eslint&quot;: &quot;true&quot;,
49
- &quot;node.js.detected.package.tslint&quot;: &quot;true&quot;,
50
- &quot;node.js.selected.package.eslint&quot;: &quot;(autodetect)&quot;,
51
- &quot;node.js.selected.package.tslint&quot;: &quot;(autodetect)&quot;,
52
- &quot;vue.rearranger.settings.migration&quot;: &quot;true&quot;
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="1213000" />
 
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