Julian Bilcke
commited on
Commit
·
9cdaa70
1
Parent(s):
217d567
1080p should also use 720p data when there is no choice
Browse files- vms/config.py +4 -0
vms/config.py
CHANGED
|
@@ -492,6 +492,10 @@ HD_TRAINING_BUCKETS = [
|
|
| 492 |
|
| 493 |
# For 1920x1080 images and videos (from 1 frame up to 272)
|
| 494 |
FHD_TRAINING_BUCKETS = [
|
|
|
|
|
|
|
|
|
|
|
|
|
| 495 |
(NB_FRAMES_1, FHD_16_9_H, FHD_16_9_W), # 1
|
| 496 |
(NB_FRAMES_9, FHD_16_9_H, FHD_16_9_W), # 8 + 1
|
| 497 |
(NB_FRAMES_17, FHD_16_9_H, FHD_16_9_W), # 16 + 1
|
|
|
|
| 492 |
|
| 493 |
# For 1920x1080 images and videos (from 1 frame up to 272)
|
| 494 |
FHD_TRAINING_BUCKETS = [
|
| 495 |
+
# since not all videos might be in 1080p,
|
| 496 |
+
# we also integrate the 720p buckets in order to increase the quality
|
| 497 |
+
*HD_TRAINING_BUCKETS,
|
| 498 |
+
|
| 499 |
(NB_FRAMES_1, FHD_16_9_H, FHD_16_9_W), # 1
|
| 500 |
(NB_FRAMES_9, FHD_16_9_H, FHD_16_9_W), # 8 + 1
|
| 501 |
(NB_FRAMES_17, FHD_16_9_H, FHD_16_9_W), # 16 + 1
|