blob: a11198319d35636df89d1612c6d265e350e724dd (
plain)
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
|
#!/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;
|
