blob: 7805247e1185084b487927d33c3286accc53671c (
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
76
77
78
79
80
81
|
#!/usr/bin/perl
use strict;
use Getopt::Long;
use IO::Handle;
STDERR->autoflush(1);
STDOUT->autoflush(1);
my $sequence_dir;
my $ab_dir;
my $dt;
my $wait;
my $buzz;
GetOptions (
"sequence_dir" => \$sequence_dir,
"ab_dir" => \$ab_dir,
"dt" => \$dt,
)
print $sequence_dir . "\n";
if (! $sequence_dir || ! $ab_dir || ! -e $sequence_dir) {
print "usage: --sequence_dir ... --ab_dir ... --dt 1\n";
exit(1);
}
if (! $dt) {
$dt = 1;
}
$sequence_dir .= "/";
$ab_dir .= "/";
mkdir($ab_dir);
mkdir($ab_dir . "A/");
mkdir($ab_dir . "B/");
mkdir($ab_dir . "A/train/");
mkdir($ab_dir . "B/train/");
mkdir($ab_dir . "A/val/");
mkdir($ab_dir . "B/val/");
mkdir($ab_dir . "A/test/");
mkdir($ab_dir . "B/test/");
our @files = ();
opendir DIR, $sequence_dir;
while (readdir DIR) {
next if /^\./;
push(@files, $_);
}
closedir DIR;
our @sequence = sort @files;
my $count = scalar(@sequence) - $dt;
print "$count images in sequence\n";
my $i; my $x; my $y; my $dir; my $id;
for ($i = 0; $i < $count; $i++) {
$id = $i;
if ( ($id % 10) == 5) {
$dir = "test/";
} elsif ( ($id % 10) == 8) {
$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 = $sequence_dir . $sequence[$id];
my $b_frame = $sequence_dir . $sequence[$id+$dt];
my $fn = sprintf("frame_%05d.png", $i);
#print "$fn $x $y $a_frame\n";
system("ln", "-s", $a_frame, $ab_dir . "A/" . $dir . $fn);
system("ln", "-s", $b_frame, $ab_dir . "B/" . $dir . $fn);
}
print $count . "\n";
|
