new builder scripts

1 files changed, 75 insertions, 0 deletions

diff --git a/scripts/builders/flat-multi-dataset.pl b/scripts/builders/flat-multi-dataset.pl
new file mode 100755
index 0000000..a111983
--- /dev/null
+++ b/scripts/builders/flat-multi-dataset.pl
@@ -0,0 +1,75 @@
+#!/usr/bin/perl
+
+use strict;
+
+# Processing script for Adversarial Video Generation library
+# Images should all be 320x or similar
+
+our $images_dir = "/home/lens/Desktop/woods_gray/";
+our $thumbs_dir = "/home/lens/Desktop/thumbs/kaulsdorf_gray/";
+
+mkdir($thumbs_dir);
+mkdir($thumbs_dir . "Train/");
+mkdir($thumbs_dir . "Test/");
+
+our $frames_per_dir = 256;
+our $test_split = 5; # 1/N videos will be for the test split
+
+our $dircount = 0;
+
+sub parse_dir ($) {
+  our ($im_dir) = @_;
+  our @files = ();
+
+  print $im_dir . "\n";
+
+  opendir DIR, $im_dir;
+  while (readdir DIR) {
+    next if /^\./;
+    push(@files, $_);
+  }
+  closedir DIR;
+
+  our @images = sort @files;
+
+  my $count = scalar(@images);
+
+  my $i;
+  my $j;
+  my $dirid = -2;
+  my $nextid;
+  my $dir;
+  for ($i = 0; $i < $count; $i++) {
+    $nextid = int($i / $frames_per_dir) + $dircount;
+    if ($nextid != $dirid) {
+      $dirid = $nextid;
+      if (($dirid % $test_split) != ($test_split-1)) {
+        $dir = $thumbs_dir . "Train/video" . $dirid . "/";
+      } else {
+        $dir = $thumbs_dir . "Test/video" . $dirid . "/";
+      }
+      print $dir . "\n";
+      mkdir($dir);
+      $j = 0;
+    } else {
+      $j += 1;
+    }
+
+    my $src = $im_dir . $images[$i];
+    my $dst = $dir . sprintf("frame%03d.png", $j);
+    system("cp", $src, $dst);
+    # print $dst . "\n";
+  }
+
+  $dircount = $nextid + 1;
+  print $count . " " . $dircount . "\n";
+}
+
+opendir D, $images_dir;
+while (readdir D) {
+  next if /^\./;
+  next if ! -d $images_dir . $_;
+  parse_dir($images_dir . $_ . "/");
+}
+closedir D;
+