jbilcke-hf HF Staff commited on
Commit
9cdaa70
·
1 Parent(s): 217d567

1080p should also use 720p data when there is no choice

Browse files
Files changed (1) hide show
  1. 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