diff options
Diffstat (limited to 'scripts/builders/pair-dataset.pl')
| -rw-r--r-- | scripts/builders/pair-dataset.pl | 23 |
1 files changed, 16 insertions, 7 deletions
diff --git a/scripts/builders/pair-dataset.pl b/scripts/builders/pair-dataset.pl index 06c40c0..100cf9f 100644 --- a/scripts/builders/pair-dataset.pl +++ b/scripts/builders/pair-dataset.pl @@ -2,8 +2,8 @@ use strict; -our $images_dir = "/home/ubuntu/Desktop/wood/"; -our $thumbs_dir = "/home/ubuntu/Desktop/thumbs/wood/"; +our $images_dir = "/home/lens/Desktop/messi/"; +our $thumbs_dir = "/home/lens/Desktop/thumbs/messi/"; our $dt = 1; @@ -11,8 +11,7 @@ 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/train/"); mkdir($thumbs_dir . "A/val/"); mkdir($thumbs_dir . "B/val/"); mkdir($thumbs_dir . "A/test/"); mkdir($thumbs_dir . "B/test/"); @@ -28,7 +27,14 @@ closedir DIR; our @images = sort @files; -my $count = scalar(@images) - $dt; +my $count = 3500; # scalar(@images) - $dt; + +# 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++) { @@ -44,11 +50,14 @@ for ($i = 0; $i < $count; $i++) { 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 $fn = sprintf("frame_%05d.png", $i); + 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, "-resize", '256x256!', '-canny', '0x1+10%+30%', $thumbs_dir . "A/" . $dir . $fn); + system("convert", $a_frame, "-resize", '256x256!', $thumbs_dir . "A/" . $dir . $fn); system("convert", $b_frame, "-resize", '256x256!', $thumbs_dir . "B/" . $dir . $fn); } print $count . "\n"; |