Skip to content

Commit 5a5d330

Browse files
lukaszkaiserlukaszkaiser
authored
Merge pull request tensorflow#2969 from coreylynch/master
Adding TCN.
2 parents 69cf6fc + aa3d442 commit 5a5d330

39 files changed

+6734
-0
lines changed

‎CODEOWNERS‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ research/slim/* @sguada @nathansilberman
3232
research/street/* @theraysmith
3333
research/swivel/* @waterson
3434
research/syntaxnet/* @calberti @andorardo @bogatyy @markomernick
35+
research/tcn/* @coreylynch @sermanet
3536
research/textsum/* @panyx0718 @peterjliu
3637
research/transformer/* @daviddao
3738
research/video_prediction/* @cbfinn

‎research/README.md‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ installation](https://www.tensorflow.org/install).
6161
using a Deep RNN.
6262
- [swivel](swivel): the Swivel algorithm for generating word embeddings.
6363
- [syntaxnet](syntaxnet): neural models of natural language syntax.
64+
- [tcn](tcn): Self-supervised representation learning from multi-view video.
6465
- [textsum](textsum): sequence-to-sequence with attention model for text
6566
summarization.
6667
- [transformer](transformer): spatial transformer network, which allows the

‎research/tcn/BUILD‎

Lines changed: 213 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,213 @@
1+
package(default_visibility = [":internal"])
2+
3+
licenses(["notice"]) # Apache 2.0
4+
5+
exports_files(["LICENSE"])
6+
7+
package_group(
8+
name = "internal",
9+
packages = [
10+
"//tcn/...",
11+
],
12+
)
13+
14+
py_binary(
15+
name = "download_pretrained",
16+
srcs = [
17+
"download_pretrained.py",
18+
],
19+
)
20+
21+
py_binary(
22+
name = "generate_videos",
23+
srcs = [
24+
"generate_videos.py",
25+
],
26+
main = "generate_videos.py",
27+
deps = [
28+
":data_providers",
29+
":get_estimator",
30+
":util",
31+
],
32+
)
33+
34+
py_test(
35+
name = "svtcn_loss_test",
36+
size = "medium",
37+
srcs = [
38+
"estimators/svtcn_loss.py",
39+
"estimators/svtcn_loss_test.py",
40+
],
41+
deps = [
42+
":util",
43+
],
44+
)
45+
46+
py_library(
47+
name = "data_providers",
48+
srcs = [
49+
"data_providers.py",
50+
],
51+
deps = [
52+
":preprocessing",
53+
],
54+
)
55+
56+
py_test(
57+
name = "data_providers_test",
58+
size = "large",
59+
srcs = ["data_providers_test.py"],
60+
deps = [
61+
":data_providers",
62+
],
63+
)
64+
65+
py_library(
66+
name = "preprocessing",
67+
srcs = [
68+
"preprocessing.py",
69+
],
70+
)
71+
72+
py_binary(
73+
name = "get_estimator",
74+
srcs = [
75+
"estimators/get_estimator.py",
76+
],
77+
deps = [
78+
":mvtcn_estimator",
79+
":svtcn_estimator",
80+
],
81+
)
82+
83+
py_binary(
84+
name = "base_estimator",
85+
srcs = [
86+
"estimators/base_estimator.py",
87+
"model.py",
88+
],
89+
deps = [
90+
":data_providers",
91+
":util",
92+
],
93+
)
94+
95+
py_library(
96+
name = "util",
97+
srcs = [
98+
"utils/luatables.py",
99+
"utils/progress.py",
100+
"utils/util.py",
101+
],
102+
)
103+
104+
py_binary(
105+
name = "mvtcn_estimator",
106+
srcs = [
107+
"estimators/mvtcn_estimator.py",
108+
],
109+
deps = [
110+
":base_estimator",
111+
],
112+
)
113+
114+
py_binary(
115+
name = "svtcn_estimator",
116+
srcs = [
117+
"estimators/svtcn_estimator.py",
118+
"estimators/svtcn_loss.py",
119+
],
120+
deps = [
121+
":base_estimator",
122+
],
123+
)
124+
125+
py_binary(
126+
name = "train",
127+
srcs = [
128+
"train.py",
129+
],
130+
deps = [
131+
":data_providers",
132+
":get_estimator",
133+
":util",
134+
],
135+
)
136+
137+
py_binary(
138+
name = "labeled_eval",
139+
srcs = [
140+
"labeled_eval.py",
141+
],
142+
deps = [
143+
":get_estimator",
144+
],
145+
)
146+
147+
py_test(
148+
name = "labeled_eval_test",
149+
size = "small",
150+
srcs = ["labeled_eval_test.py"],
151+
deps = [
152+
":labeled_eval",
153+
],
154+
)
155+
156+
py_binary(
157+
name = "eval",
158+
srcs = [
159+
"eval.py",
160+
],
161+
deps = [
162+
":get_estimator",
163+
],
164+
)
165+
166+
py_binary(
167+
name = "alignment",
168+
srcs = [
169+
"alignment.py",
170+
],
171+
deps = [
172+
":get_estimator",
173+
],
174+
)
175+
176+
py_binary(
177+
name = "visualize_embeddings",
178+
srcs = [
179+
"visualize_embeddings.py",
180+
],
181+
deps = [
182+
":data_providers",
183+
":get_estimator",
184+
":util",
185+
],
186+
)
187+
188+
py_binary(
189+
name = "webcam",
190+
srcs = [
191+
"dataset/webcam.py",
192+
],
193+
main = "dataset/webcam.py",
194+
)
195+
196+
py_binary(
197+
name = "images_to_videos",
198+
srcs = [
199+
"dataset/images_to_videos.py",
200+
],
201+
main = "dataset/images_to_videos.py",
202+
)
203+
204+
py_binary(
205+
name = "videos_to_tfrecords",
206+
srcs = [
207+
"dataset/videos_to_tfrecords.py",
208+
],
209+
main = "dataset/videos_to_tfrecords.py",
210+
deps = [
211+
":preprocessing",
212+
],
213+
)

0 commit comments

Comments
 (0)