-
Notifications
You must be signed in to change notification settings - Fork 2
/
dvc.lock
499 lines (499 loc) · 17 KB
/
dvc.lock
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
schema: '2.0'
stages:
download-images:
cmd: python scripts/download-images.py datasets/watch-faces.json --concurrent
deps:
- path: datasets/watch-faces.json
md5: 1b5a5c7576bbc6aab7d84fa337d40804
size: 1895716
- path: scripts/download-images.py
md5: a037cfab1a482a356f1d7d62804041d3
size: 3961
outs:
- path: datasets/test
md5: c3b9fc1ebd7a05cbf2dfe0c54f2fe397.dir
size: 18869343
nfiles: 55
- path: datasets/train
md5: 564e5c2bd6e82f4d17952f9fc2d0dd8b.dir
size: 220071018
nfiles: 692
- path: datasets/val
md5: c2e5983ceddfc4ca034e7992c059f2fd.dir
size: 20206227
nfiles: 42
- path: datasets/watch-faces-local.json
md5: 5967d4bbad75ce2f0becf98f53c2029c
size: 1877638
train-detector:
cmd: python watch_recognition/watch_recognition/train/object_detection_tf_model_garden.py
"train-configs/tf-model-garden/watch-face-detector/" --seed 42
deps:
- path: datasets/tf-records/object-detection/watch-faces/
md5: 45c36d0d23bc8b31f6038bcf96bdec66.dir
size: 259463443
nfiles: 4
- path: train-configs/tf-model-garden/watch-face-detector/retinanet_task.yaml
md5: 4cc3a22904a608cb4a5e0bc19f2bdfad
size: 3247
- path: train-configs/tf-model-garden/watch-face-detector/runtime.yaml
md5: 71503a0285b578c4ccd9b2cbea2f4a74
size: 405
- path: train-configs/tf-model-garden/watch-face-detector/trainer.yaml
md5: d6a17c825eeee10d90ca143bb4604f69
size: 845
params:
params.yaml:
detector:
epochs: 100
batch-size: 1
confidence-threshold: 0.5
label_to_cls:
WatchFace: 1
seed: 42
outs:
- path: debug/detector/
md5: d70b5004a852f4767a7bf2587f5fbe80.dir
size: 272870
nfiles: 1
- path: models/detector/
md5: 58518bc50561568de3f98686f5935ec9.dir
size: 321639160
nfiles: 5
eval-detector:
cmd: python watch_recognition/watch_recognition/eval/object_detection_eval.py
deps:
- path: datasets/test
md5: c3b9fc1ebd7a05cbf2dfe0c54f2fe397.dir
size: 18869343
nfiles: 55
- path: datasets/train
md5: 564e5c2bd6e82f4d17952f9fc2d0dd8b.dir
size: 220071018
nfiles: 692
- path: datasets/val
md5: c2e5983ceddfc4ca034e7992c059f2fd.dir
size: 20206227
nfiles: 42
- path: exported_models/detector/serving/
md5: 65386a7d39f85d1f17cad3295698eb35.dir
size: 143112069
nfiles: 8
outs:
- path: example_predictions/detector/train_0.jpg
md5: 72eb85d4b248b013e1780d962d36ef55
size: 50754
- path: example_predictions/detector/train_1.jpg
md5: c9adb048c99f16a451cc270c8b05db64
size: 44144
- path: example_predictions/detector/train_2.jpg
md5: 423b92d7f8ee6ada84c781025ac51784
size: 43900
- path: example_predictions/detector/val_0.jpg
md5: 0e3ea39f90c009e79c0daa1a7dad560e
size: 40986
- path: example_predictions/detector/val_1.jpg
md5: 040d6f52b16539fccd2ba9fd33110de5
size: 55781
- path: example_predictions/detector/val_2.jpg
md5: b55dc9acd736b7b06cb419a2e6f4e466
size: 30102
- path: metrics/detector/train/PR-IoU@0.50.tsv
md5: 15f501a877d062e3be3477aab9af5629
size: 2423
- path: metrics/detector/train/PR-IoU@0.75.tsv
md5: d93af0ad79d3a1a59493240a36f3bd90
size: 2455
- path: metrics/detector/train/PR-IoU@0.95.tsv
md5: e0209a837a72ba9eb6fc24ae8d0512ed
size: 1399
- path: metrics/detector/train/coco.json
md5: 073139f95b83e099106fef742dacf704
size: 286
- path: metrics/detector/train/detection.json
md5: c327e36cd737ce141c1b92b15e9bd244
size: 135
- path: metrics/detector/val/PR-IoU@0.50.tsv
md5: 29a651c0d7c2a10b5f24de57838a612c
size: 2164
- path: metrics/detector/val/PR-IoU@0.75.tsv
md5: 6053e7e25f73c7ce5c3924fa9f1aff05
size: 1750
- path: metrics/detector/val/PR-IoU@0.95.tsv
md5: ec06f56e8221d92b3a41680ab75c418d
size: 1059
- path: metrics/detector/val/coco.json
md5: fdeb918be2d64507433287b4e30b7f8a
size: 285
- path: metrics/detector/val/detection.json
md5: 59b8478f36d6aeeba97fcedf3d44902f
size: 133
update-metrics:
cmd: python scripts/update-metrics-table-and-graph.py
deps:
- path: metrics
md5: 7720dbd530419e0aed0127b256d6495e.dir
size: 212311
nfiles: 38
- path: scripts/update-metrics-table-and-graph.py
md5: 6a91ab9d3e3ba8ec6a3fa252dd1853cb
size: 2068
train-keypoint:
cmd: python watch_recognition/watch_recognition/train/heatmap_regression_task.py
--epochs 100 --batch-size 32 --confidence-threshold 0.5 --seed 42
deps:
- path: datasets/test
md5: 0aeb029db40e49068d63308665810b57.dir
size: 18701099
nfiles: 54
- path: datasets/train
md5: 9c414afa46808b376366ae8e0ff3c786.dir
size: 170152358
nfiles: 481
- path: datasets/val
md5: 1ab506f4b22a797ed4afeaf195c51466.dir
size: 19774777
nfiles: 40
- path: datasets/watch-faces-local.json
md5: ddb1e52cb640b1af3e8ba6c7c85c943a
size: 1744824
- path: watch_recognition/watch_recognition/train/heatmap_regression_task.py
md5: f6a86273f89a38a8abdb8575ec154e9d
size: 8953
params:
params.yaml:
keypoint:
epochs: 100
batch-size: 32
confidence-threshold: 0.5
label_to_cls:
Top: 0
Center: 1
Crown: 2
disk_radius: 5
max_images:
seed: 42
outs:
- path: debug/keypoint/
md5: 2ac270cd02a60722351bada220d03d8c.dir
size: 399737
nfiles: 2
- path: example_predictions/keypoint/test-image-2.jpg
md5: 89d44b8cb435b07ac4634e13eaa00860
size: 66334
- path: metrics/keypoint/metrics.json
md5: c479e42ab5fdf1aa00041a9ea746728e
size: 214
- path: metrics/keypoint/plots/
md5: c6572a9c857b01e34dc8b7b1bc49ff3c.dir
size: 8908
nfiles: 4
- path: models/keypoint/
md5: 4a34792d0d5209c621c38464f1614818.dir
size: 35988621
nfiles: 6
eval-keypoint:
cmd: python watch_recognition/watch_recognition/eval/keypoint_detection_eval.py
--kp-confidence-threshold 0.5
deps:
- path: models/detector/
md5: 1fab3506112358531d4e67f67318d577.dir
size: 432232708
nfiles: 5
- path: models/keypoint/
md5: 61e3f9286fc1eb9530e2878629990abc.dir
size: 35964730
nfiles: 6
- path: watch_recognition/watch_recognition/eval/keypoint_detection_eval.py
md5: 7aee96c16f50bc522e62bc9138f49247
size: 6302
outs:
- path: example_predictions/keypoint/train_0.jpg
md5: 065902c4224d6e4ded2ffd5eb1c5fb2d
size: 25458
- path: example_predictions/keypoint/train_1.jpg
md5: 87be1633d4e2629e598184dbafd5dc90
size: 21951
- path: example_predictions/keypoint/train_2.jpg
md5: f092b0e3656b1527980d3f5ee4472c4a
size: 42166
- path: example_predictions/keypoint/train_3.jpg
md5: b0e9da65d6eb0f0d881d395ab00daae6
size: 38041
- path: example_predictions/keypoint/train_4.jpg
md5: 6ec1e284d155bb0795063726e8ef81af
size: 37605
- path: example_predictions/keypoint/val_0.jpg
md5: f4233d485b75ab4fa2c6ddf7b09af9d7
size: 53245
- path: example_predictions/keypoint/val_1.jpg
md5: f3872f5c07a702cfd6667dc2d399fc21
size: 60953
- path: example_predictions/keypoint/val_2.jpg
md5: e688033fb3d853f1812251060dca598b
size: 48369
- path: example_predictions/keypoint/val_3.jpg
md5: 571341fd9f985c0e7be89ea985e9fe93
size: 57673
- path: example_predictions/keypoint/val_4.jpg
md5: 3ede96a954d7b20bb21a9a9f2a4d8621
size: 51549
- path: metrics/keypoint/coco_train.json
md5: 4c4fdf89b72a329b4e0b47985bf12a9e
size: 260
- path: metrics/keypoint/coco_val.json
md5: fad55e8d91734f2d590a026e55746447
size: 248
train-segmentation:
cmd: python watch_recognition/watch_recognition/train/segmentation_task.py --epochs
150 --batch-size 32 --confidence-threshold 0.5 --seed 42
deps:
- path: datasets/test
md5: 0aeb029db40e49068d63308665810b57.dir
size: 18701099
nfiles: 54
- path: datasets/train
md5: 9c414afa46808b376366ae8e0ff3c786.dir
size: 170152358
nfiles: 481
- path: datasets/val
md5: 1ab506f4b22a797ed4afeaf195c51466.dir
size: 19774777
nfiles: 40
- path: datasets/watch-faces-local.json
md5: ddb1e52cb640b1af3e8ba6c7c85c943a
size: 1744824
- path: watch_recognition/watch_recognition/train/segmentation_task.py
md5: af4c58046894343dd442a39bbfbd6411
size: 9265
params:
params.yaml:
seed: 42
segmentation:
epochs: 150
batch-size: 32
confidence-threshold: 0.5
label_to_cls:
Hands: 0
bbox_labels:
- WatchFace
outs:
- path: debug/segmentation/
md5: 78f6047cbb06c4e7889f881f7143064d.dir
size: 761590
nfiles: 4
- path: example_predictions/segmentation/test-image-2.jpg
md5: 44fc2994fb44e280363c023622c24163
size: 64278
- path: metrics/segmentation/metrics.json
md5: ed15dc7ca524e01ef350507b0f8cda43
size: 215
- path: metrics/segmentation/plots/
md5: e0203589dc6763f0c85af5d7f2eee974.dir
size: 13572
nfiles: 4
- path: models/segmentation/
md5: 986e39bc0071d95599ee1918c760865c.dir
size: 35949342
nfiles: 5
eval-end-2-end:
cmd: python watch_recognition/watch_recognition/eval/end_to_end_eval.py --run-concurrently
--split=val
deps:
- path: datasets/watch-faces-local.json
md5: b70283f3027b17d5c33f72fe0685e3a5
size: 1573168
- path: models/detector
md5: 1fab3506112358531d4e67f67318d577.dir
size: 432232708
nfiles: 5
- path: models/keypoint
md5: 61e3f9286fc1eb9530e2878629990abc.dir
size: 35964730
nfiles: 6
- path: models/segmentation
md5: e603288dbbd9abdd9725e917d2b4ff50.dir
size: 35949358
nfiles: 5
- path: watch_recognition/watch_recognition/eval/end_to_end_eval.py
md5: c48df0afed288ec1b487fc2088b43b86
size: 8652
outs:
- path: metrics/end_2_end_eval.csv
md5: 760a2cbeead50d05ac8874d97761747a
size: 5955
- path: metrics/end_2_end_summary.json
md5: efa5cc3780b031a247ae9dc26e7daa39
size: 132
eval-segmentation:
cmd: python watch_recognition/watch_recognition/eval/segmentation_eval.py --confidence-threshold
0.5
deps:
- path: models/detector/
md5: c4ac10282d0de69b98c3d761036bb86a.dir
size: 1427658959
nfiles: 18
- path: models/segmentation/
md5: 986e39bc0071d95599ee1918c760865c.dir
size: 35949342
nfiles: 5
- path: watch_recognition/watch_recognition/eval/segmentation_eval.py
md5: bd04e7ffd7d9d64e4258c38f704171d9
size: 3342
outs:
- path: example_predictions/segmentation/train_0.jpg
md5: 95e5b350ac207e3e577a162d52342736
size: 32390
- path: example_predictions/segmentation/train_1.jpg
md5: 5aac5474b3aa20d54dff0e164799248a
size: 45347
- path: example_predictions/segmentation/train_2.jpg
md5: 806587e18fe1e996c859115c119c3d60
size: 33351
- path: example_predictions/segmentation/train_3.jpg
md5: 240f88c937e53319fa7ded3cc020fd66
size: 37194
- path: example_predictions/segmentation/train_4.jpg
md5: 731db63f31e1151de671413f8a50dcb7
size: 42616
- path: example_predictions/segmentation/val_0.jpg
md5: a8b32c9a77b262f766253bc28275a2aa
size: 44631
- path: example_predictions/segmentation/val_1.jpg
md5: 31f47e2b4b712aaf3becf2a8c5ed6303
size: 56784
- path: example_predictions/segmentation/val_2.jpg
md5: 51d856ab51eaf02d863df7af25c1bc8c
size: 42611
- path: example_predictions/segmentation/val_3.jpg
md5: 9f56190150349dd382695b641db57049
size: 52859
- path: example_predictions/segmentation/val_4.jpg
md5: 5160229796dbe2b20b96c4579661bf1a
size: 48967
generate-detection-dataset:
cmd: python watch_recognition/watch_recognition/datasets/generate_detection_tf_records.py datasets/watch-faces-local.json datasets/tf-records/object-detection/watch-faces/ --num-shards=1
--run-concurrently
deps:
- path: datasets/test
md5: c3b9fc1ebd7a05cbf2dfe0c54f2fe397.dir
size: 18869343
nfiles: 55
- path: datasets/train
md5: 564e5c2bd6e82f4d17952f9fc2d0dd8b.dir
size: 220071018
nfiles: 692
- path: datasets/val
md5: c2e5983ceddfc4ca034e7992c059f2fd.dir
size: 20206227
nfiles: 42
- path: datasets/watch-faces-local.json
md5: 5967d4bbad75ce2f0becf98f53c2029c
size: 1877638
params:
params.yaml:
detector.label_to_cls:
WatchFace: 1
outs:
- path: datasets/tf-records/object-detection/watch-faces/watch-faces-test-00001-of-00001.tfrecord
md5: a555b808d9d7f608d3204f5aa90b02eb
size: 18894906
- path: datasets/tf-records/object-detection/watch-faces/watch-faces-train-00001-of-00001.tfrecord
md5: ebdb72874bb6c7f193fc589f736bc8e3
size: 220347192
- path: datasets/tf-records/object-detection/watch-faces/watch-faces-val-00001-of-00001.tfrecord
md5: 61b08868ea0f804e6fd9dcaaee60055a
size: 20220589
generate-watch-hands-dataset:
cmd: python watch_recognition/watch_recognition/datasets/generate_segmentation_tf_records.py
datasets/watch-faces-local.json datasets/tf-records/segmentation/watch-hands/ --num-shards=1
--run-concurrently
deps:
- path: datasets/test
md5: 0aeb029db40e49068d63308665810b57.dir
size: 18701099
nfiles: 54
- path: datasets/val
md5: 1ab506f4b22a797ed4afeaf195c51466.dir
size: 19774777
nfiles: 40
- path: datasets/watch-faces-local.json
md5: ddb1e52cb640b1af3e8ba6c7c85c943a
size: 1744824
params:
params.yaml:
segmentation.bbox_labels:
- WatchFace
segmentation.label_to_cls:
Hands: 1
outs:
- path: datasets/tf-records/segmentation/watch-hands/watch-hands-train-00001-of-00001.tfrecord
md5: 2a101ccdbac32084074ed1dea1f0da4b
size: 26611279
- path: datasets/tf-records/segmentation/watch-hands/watch-hands-val-00001-of-00001.tfrecord
md5: a0e242785798e143ed2c1cb8ea08812a
size: 2716719
export-detector:
cmd: python watch_recognition/watch_recognition/export/export_retinanet.py models/detector/
deps:
- path: datasets/tf-records/object-detection/watch-faces/watch-faces-train-00001-of-00001.tfrecord
md5: ebdb72874bb6c7f193fc589f736bc8e3
size: 220347192
- path: models/detector/
md5: 58518bc50561568de3f98686f5935ec9.dir
size: 321639160
nfiles: 5
- path: train-configs/tf-model-garden/watch-face-detector/retinanet_task.yaml
md5: 4cc3a22904a608cb4a5e0bc19f2bdfad
size: 3247
- path: train-configs/tf-model-garden/watch-face-detector/runtime.yaml
md5: 71503a0285b578c4ccd9b2cbea2f4a74
size: 405
- path: train-configs/tf-model-garden/watch-face-detector/trainer.yaml
md5: d6a17c825eeee10d90ca143bb4604f69
size: 845
outs:
- path: example_predictions/detector/IMG_0040.jpg
md5: 5abcc4ff8c0980a4e3f549c0350164f3
size: 229262
- path: exported_models/detector/lite16/assets.extra/tf_serving_warmup_requests
md5: 8c7059090db70976a01dfc2e8dd1c968
size: 1769557
- path: exported_models/detector/lite16/model.tflite
md5: 6e4ece3f68f8dbdc8539701d2157f0ad
size: 54973072
- path: exported_models/detector/lite8/assets.extra/tf_serving_warmup_requests
md5: 8c7059090db70976a01dfc2e8dd1c968
size: 1769557
- path: exported_models/detector/lite8/model.tflite
md5: 1c42bf583af00e08292d6eeca93530e4
size: 91696144
- path: exported_models/detector/serving/assets.extra/tf_serving_warmup_requests
md5: 783b65cb6c06d334aaf972183d2c4409
size: 11520090
- path: exported_models/detector/serving/fingerprint.pb
md5: 96646ded012898254be320f2d8b65943
size: 55
- path: exported_models/detector/serving/saved_model.pb
md5: bec4b4e548ae57f5a2230e93c11a4953
size: 23869403
- path: exported_models/detector/serving/variables/variables.data-00000-of-00001
md5: 47de8ee9f3fdf79eb71ad04f26c6d272
size: 107671904
- path: exported_models/detector/serving/variables/variables.index
md5: c44f4054fe1df9b9eb6374d50816c1e0
size: 50501
render-demo:
cmd: scripts/render-demo-movie.py example_data/IMG_1200_720p.mov demo/demo.mp4
--enable-multithreading
deps:
- path: example_data/IMG_1200_720p.mov
md5: 5dfad65d6cb1231b1dc70d24153ef52a
size: 12545634
- path: scripts/render-demo-movie.py
md5: 7ca4659bab16040db84a1f87bbc67385
size: 4462
outs:
- path: demo/demo.mp4
md5: 2a53a37aa96b97c95348d08063e4809e
size: 8935751