blob: bc59723bfa00a1e5ddbe95e92cf7d13061e20a73 (
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
|
#!/usr/bin/perl
use strict;
our $dt = 4;
our $count = 30000;
our $images_dir = "woodtripod/";
our $thumbs_dir = "thumbs/randomcrops_" . $count . "_" . $dt . "/";
mkdir($thumbs_dir);
mkdir($thumbs_dir . "A/");
mkdir($thumbs_dir . "B/");
mkdir($thumbs_dir . "A/train/");
mkdir($thumbs_dir . "B/train/");
mkdir($thumbs_dir . "A/val/");
mkdir($thumbs_dir . "B/val/");
mkdir($thumbs_dir . "A/test/");
mkdir($thumbs_dir . "B/test/");
our @files = ();
opendir DIR, $images_dir;
while (readdir DIR) {
next if /^\./;
push(@files, $_);
}
closedir DIR;
our @images = sort @files;
# assume all images are same size
our $w = 1920;
our $h = 1080;
our $dw = 256;
our $dh = 256;
my $i; my $x; my $y; my $dir;
for ($i = 0; $i < $count; $i++) {
my $id = int($i/$count * (scalar @images - $dt));
if ( ($id % 7) == 3) {
$dir = "test/";
} elsif ( ($id % 7) == 6) {
$dir = "val/";
} else {
$dir = "train/";
}
if ( $i && ($i % 1000) == 0) {
print($id . "...\n")
}
my $x = int(rand($w - $dw));
my $y = int(rand($h - $dh));
my $a_frame = $images_dir . $images[$id];
my $b_frame = $images_dir . $images[$id+$dt];
my $crop = $dw . "x" . $dh . "+" . $x . "x" . $y;
my $fn = sprintf("frame_%04d.png", $i);
print "$fn $x $y $a_frame\n";
system("convert", $a_frame, "-crop", $crop, $thumbs_dir . "A/" . $dir . $fn);
system("convert", $b_frame, "-crop", $crop, $thumbs_dir . "B/" . $dir . $fn);
}
print $count . "\n";
|
