diff --git a/README b/README deleted file mode 100644 index e69de29b..00000000 diff --git a/readme.md b/README.md similarity index 83% rename from readme.md rename to README.md index f5161395..62200976 100644 --- a/readme.md +++ b/README.md @@ -1,6 +1,8 @@ Steps to get OpenDroneMap running: ================================== +(Requires Ubuntu 12.04 32-bit at this time. May also work with Ubuntu 12.04 64-bit) + Run install.sh to build. ``` ./install.sh ``` @@ -26,3 +28,7 @@ From Meshlab 1.3.3: * Filters:Remeshing, Simplification and Reconstruction:Surface Reconstruction Poisson * Texture the mesh * Parameterization + texturing from registered rasters + +--- + +Example data can be found at https://github.com/OpenDroneMap/odm_data diff --git a/install-centos.sh b/install-centos.sh deleted file mode 100644 index cd7e1167..00000000 --- a/install-centos.sh +++ /dev/null @@ -1,342 +0,0 @@ -#!/bin/bash/bin/bash - -set -o nounset -set -o errexit - -echo -echo " created by Daniel Schwarz/daniel.schwarz@topoi.org" -echo " released under Creative Commons/CC-BY" -echo " Attribution" -echo -echo " if the script doesn't finish properly" -echo " (i.e. it doesn't print \"script finished\" at the end)" -echo " please email me the content of the logs folder" -echo -echo -echo " - script started - `date`" - - ## dest base path - TOOLS_PATH="$PWD" - - ## paths for the tools - TOOLS_BIN_PATH="$TOOLS_PATH/bin" - TOOLS_INC_PATH="$TOOLS_PATH/include" - TOOLS_LIB_PATH="$TOOLS_PATH/lib" - TOOLS_SRC_PATH="$TOOLS_PATH/src" - TOOLS_LOG_PATH="$TOOLS_PATH/logs" - TOOLS_PATCHED_PATH="$TOOLS_PATH/patched_files" - -## get sys vars -ARCH=`uname -m` -CORES=`grep -c processor /proc/cpuinfo` - - ## loacal dest paths -if [ "$ARCH" = "i686" ]; then - LIB_PATH="/lib" -fi - -if [ "$ARCH" = "x86_64" ]; then - LIB_PATH="/lib64" -fi -INC_PATH="/usr/local/include" - - ## source paths - BUNDLER_PATH="$TOOLS_SRC_PATH/bundler" - CMVS_PATH="$TOOLS_SRC_PATH/cmvs" - PMVS_PATH="$TOOLS_SRC_PATH/pmvs" - CLAPACK_PATH="$TOOLS_SRC_PATH/clapack" - VLFEAT_PATH="$TOOLS_SRC_PATH/vlfeat" - PARALLEL_PATH="$TOOLS_SRC_PATH/parallel" - PSR_PATH="$TOOLS_SRC_PATH/PoissonRecon" - GRACLUS_PATH="$TOOLS_SRC_PATH/graclus" - - ## executables - EXTRACT_FOCAL="$TOOLS_BIN_PATH/extract_focal.pl" - MATCHKEYS="$TOOLS_BIN_PATH/KeyMatch" - MATCHKEYSFULL="$TOOLS_BIN_PATH/KeyMatchFull" - BUNDLER="$TOOLS_BIN_PATH/bundler" - BUNDLE2PVMS="$TOOLS_BIN_PATH/Bundle2PMVS" - CMVS="$TOOLS_BIN_PATH/cmvs" - PMVS="$TOOLS_BIN_PATH/pmvs2" - GENOPTION="$TOOLS_BIN_PATH/genOption" - VLSIFT="$TOOLS_BIN_PATH/vlsift" - PARALLEL="$TOOLS_BIN_PATH/parallel" - PSR="$TOOLS_BIN_PATH/PoissonRecon" - VLSIFT_TO_LOWESIFT="$TOOLS_BIN_PATH/convert_vlsift_to_lowesift.pl" - -## prevents different (localized) output -LC_ALL=C - -## removing old stuff -sudo rm -Rf "$TOOLS_BIN_PATH" -sudo rm -Rf "$TOOLS_INC_PATH" -sudo rm -Rf "$TOOLS_LIB_PATH" -sudo rm -Rf "$TOOLS_SRC_PATH" -sudo rm -Rf "$TOOLS_LOG_PATH" - -## create needed directories -mkdir -p "$TOOLS_BIN_PATH" -mkdir -p "$TOOLS_INC_PATH" -mkdir -p "$TOOLS_LIB_PATH" -mkdir -p "$TOOLS_SRC_PATH" -mkdir -p "$TOOLS_LOG_PATH" - -## output sys info -echo "System info:" > "$TOOLS_LOG_PATH/sysinfo.txt" -uname -a > "$TOOLS_LOG_PATH/sysinfo.txt" - -## install packages -echo -echo " > installing required packages" - -echo " - updating" -sudo yum update -y > "$TOOLS_LOG_PATH/apt-get_get.log" 2>&1 - -echo " - installing" -yum install -y \ - cmake gcc gcc-c++ compat-gcc-32 compat-gcc-32-c++ gcc-gfortran perl ruby rubygems git \ - curl wget \ - unzip \ - ImageMagick jhead \ - libjpeg-devel boost-devel gsl-devel libX11-devel libXext-devel lapack-devel blas-devel \ - zlib-devel \ - opencv-devel \ - > "$TOOLS_LOG_PATH/apt-get_install.log" 2>&1 - -sudo gem install parallel > /dev/null 2>&1 - -echo " < done - `date`" - -## downloading sources -echo -echo " > getting the sources" - -## getting all archives if not already present; save them to .tmp and rename them after download -while read target source -do - if [ ! -f "$target" ] ; then - echo " - getting $source" - - curl --progress-bar --insecure --location -o "$target.tmp" "$source" - mv "$target.tmp" "$target" - echo " - finished $target" - echo - else - echo " - already downloaded $source" - fi -done < "$TOOLS_LOG_PATH/extract_$i.log" 2>&1 & -done -for i in *.tgz *.tar.gz ; do - tar xzf "$i" > "$TOOLS_LOG_PATH/extract_$i.log" 2>&1 & -done -for i in *.zip ; do - unzip "$i" > "$TOOLS_LOG_PATH/extract_$i.log" 2>&1 & -done - -wait - -mv -f graclus1.2 "$GRACLUS_PATH" -mv -f clapack-3.2.1-CMAKE "$CLAPACK_PATH" -mv -f vlfeat-0.9.13 "$VLFEAT_PATH" -mv -f bundler-v0.4-source "$BUNDLER_PATH" -mv -f parallel-20100922 "$PARALLEL_PATH" -mv -f PoissonRecon "$PSR_PATH" -mv -f cmvs "$CMVS_PATH" - -echo " < done - `date`" - - -## copying patches -echo -echo " - copying patches" -echo - -for file in `find $TOOLS_PATCHED_PATH -type f -print` ; do - cp $file $TOOLS_PATH/${file/$TOOLS_PATCHED_PATH/.} -done - -echo " < done - `date`" - - -# building -echo -echo " - building" -echo - -sudo chown -R `id -u`:`id -g` * -sudo chmod -R 777 * - - -echo " > graclus" - cd "$GRACLUS_PATH" - - if [ "$ARCH" = "i686" ]; then - sed -i "$GRACLUS_PATH/Makefile.in" -e "11c\COPTIONS = -DNUMBITS=32" - fi - - if [ "$ARCH" = "x86_64" ]; then - sed -i "$GRACLUS_PATH/Makefile.in" -e "11c\COPTIONS = -DNUMBITS=64" - fi - - echo " - cleaning graclus" - make clean > "$TOOLS_LOG_PATH/graclus_1_clean.log" 2>&1 - - echo " - building graclus" - make -j > "$TOOLS_LOG_PATH/graclus_2_build.log" 2>&1 - - mkdir "$TOOLS_INC_PATH/metisLib" - cp -f "$GRACLUS_PATH/metisLib/"*.h "$TOOLS_INC_PATH/metisLib/" - - cp -f lib* "$TOOLS_LIB_PATH/" -echo " < done - `date`" -echo - -echo " > poisson surface reconstruction " - cd "$PSR_PATH" - - sed -i "$PSR_PATH/Makefile" -e "21c\BIN = ./" - - echo " - building poisson surface reconstruction" - make -j > "$TOOLS_LOG_PATH/poisson_1_build.log" 2>&1 - - cp -f "$PSR_PATH/PoissonRecon" "$TOOLS_BIN_PATH/PoissonRecon" - -echo " < done - `date`" -echo - - -echo " > parallel" - cd "$PARALLEL_PATH" - - echo " - configuring parallel" - ./configure > "$TOOLS_LOG_PATH/parallel_1_build.log" 2>&1 - - echo " - building paralel" - make -j > "$TOOLS_LOG_PATH/parallel_2_build.log" 2>&1 - - cp -f src/parallel "$TOOLS_BIN_PATH/" - -echo " < done - `date`" -echo - - -echo " > clapack" - cd "$CLAPACK_PATH" - cp make.inc.example make.inc - - set +e - echo " - building clapack" - make all -j > "$TOOLS_LOG_PATH/clapack_1_build.log" 2>&1 - set -e - - echo " - installing clapack" - make lapack_install > "$TOOLS_LOG_PATH/clapack_2_install.log" 2>&1 - - cp -Rf INCLUDE "$INC_PATH/clapack" - -echo " < done - `date`" -echo - - -echo " > vlfeat" - cd "$VLFEAT_PATH" - - echo " - installing vlfeat" - - if [ "$ARCH" = "i686" ]; then - cp -f "$VLFEAT_PATH/bin/glnx86/sift" "$TOOLS_BIN_PATH/vlsift" - cp -f "$VLFEAT_PATH/bin/glnx86/libvl.so" "$TOOLS_LIB_PATH/" - fi - - if [ "$ARCH" = "x86_64" ]; then - cp -f "$VLFEAT_PATH/bin/glnxa64/sift" "$TOOLS_BIN_PATH/vlsift" - cp -f "$VLFEAT_PATH/bin/glnxa64/libvl.so" "$TOOLS_LIB_PATH/" - fi -echo " < done - `date`" -echo - - -echo " > cmvs" - cd "$CMVS_PATH/program/main" - - sed -i "$CMVS_PATH/program/main/genOption.cc" -e "5c\#include \n" - sed -i "$CMVS_PATH/program/base/cmvs/bundle.cc" -e "3c\#include \n" - - sed -i "$CMVS_PATH/program/main/Makefile" -e "10c\#Your INCLUDE path (e.g., -I\/usr\/include)" - sed -i "$CMVS_PATH/program/main/Makefile" -e "11c\YOUR_INCLUDE_PATH =-I$INC_PATH -I$TOOLS_INC_PATH" - sed -i "$CMVS_PATH/program/main/Makefile" -e "13c\#Your metis directory (contains header files under graclus1.2/metisLib/)" - sed -i "$CMVS_PATH/program/main/Makefile" -e "14c\YOUR_INCLUDE_METIS_PATH = -I$TOOLS_INC_PATH/metisLib/" - sed -i "$CMVS_PATH/program/main/Makefile" -e "16c\#Your LDLIBRARY path (e.g., -L/usr/lib)" - sed -i "$CMVS_PATH/program/main/Makefile" -e "17c\YOUR_LDLIB_PATH = -L$LIB_PATH -L$TOOLS_LIB_PATH" - - if [ "$ARCH" = "i686" ]; then - sed -i "$CMVS_PATH/program/main/Makefile" -e "22c\CXXFLAGS_CMVS = -O2 -Wall -Wno-deprecated -DNUMBITS=32 \\\\" - sed -i "$CMVS_PATH/program/main/Makefile" -e '24c\ -fopenmp -DNUMBITS=32 ${OPENMP_FLAG}' - fi - - if [ "$ARCH" = "x86_64" ]; then - sed -i "$CMVS_PATH/program/main/Makefile" -e "22c\CXXFLAGS_CMVS = -O2 -Wall -Wno-deprecated -DNUMBITS=64 \\\\" - sed -i "$CMVS_PATH/program/main/Makefile" -e '24c\ -fopenmp -DNUMBITS=64 ${OPENMP_FLAG}' - fi - - echo " - cleaning cmvs" - make clean > "$TOOLS_LOG_PATH/cmvs_1_clean.log" 2>&1 - - echo " - building cmvs" - make -j > "$TOOLS_LOG_PATH/cmvs_2_build.log" 2>&1 - - echo " - make depend cmvs" - sudo make depend > "$TOOLS_LOG_PATH/cmvs_3_depend.log" 2>&1 - - cp -f "$CMVS_PATH/program/main/cmvs" "$CMVS_PATH/program/main/pmvs2" "$CMVS_PATH/program/main/genOption" "$TOOLS_BIN_PATH/" - cp -f "$CMVS_PATH/program/main/"*so* "$TOOLS_LIB_PATH/" -echo " < done - `date`" -echo - - -echo " > bundler" - cd "$BUNDLER_PATH" - - sed -i "$BUNDLER_PATH/src/BundlerApp.h" -e "620c\ BundlerApp();" - - echo " - cleaning bundler" - make clean > "$TOOLS_LOG_PATH/bundler_1_clean.log" 2>&1 - - echo " - building bundler" - make -j > "$TOOLS_LOG_PATH/bundler_2_build.log" 2>&1 - - cp -f "$BUNDLER_PATH/bin/Bundle2PMVS" "$BUNDLER_PATH/bin/Bundle2Vis" "$BUNDLER_PATH/bin/KeyMatchFull" "$BUNDLER_PATH/bin/KeyMatch" "$BUNDLER_PATH/bin/bundler" "$BUNDLER_PATH/bin/RadialUndistort" "$TOOLS_BIN_PATH/" - - cp -f "$BUNDLER_PATH/lib/libANN_char.so" "$TOOLS_LIB_PATH/" -echo " < done - `date`" -echo - - -cd "$TOOLS_PATH" - -sudo install -o `id -u` -g `id -g` -m 644 -t "$LIB_PATH" lib/*.so -sudo /sbin/ldconfig -v > "$TOOLS_LOG_PATH/ldconfig.log" 2>&1 - -sudo chown -R `id -u`:`id -g` * -sudo chmod -R 777 * - -echo " - script finished - `date`" - -exit diff --git a/run.rb b/run.rb deleted file mode 100644 index e4853237..00000000 --- a/run.rb +++ /dev/null @@ -1,426 +0,0 @@ -#!/usr/bin/ruby -# encoding: utf-8 - -require 'rubygems' -require 'optparse' -require 'pp' -require 'parallel' -require 'fileutils' - -#alias :puts_original :puts -# -#def puts (*args) -# puts_original(args) -# puts_original(args) -#end - -$options = {} - -optparse = OptionParser.new do|opts| - opts.banner = "Usage: run.rb [options] " - - $options[:base_path] = "." - - $options[:resize_to] = 1200 - opts.on('-r', '--resize-to int', "") do |param| - $options[:resize_to] = param.to_i - end - - $options[:match_size] = 200 - opts.on('-m', '--match-size int', "") do |param| - $options[:match_size] = param.to_i - end - - $options[:start_with] = "resize" - opts.on('', '--start-with ', "values: \"resize\", \"getKeypoints\", \"match\", \"bundler\", \"cmvs\", \"pmvs\"") do |param| - $options[:start_with] = param - end - - $options[:end_with] = "pmvs" - opts.on('', '--end-with ', "values: \"resize\", \"getKeypoints\", \"match\", \"bundler\", \"cmvs\", \"pmvs\"") do |param| - $options[:end_with] = param - end - - $options[:force_ccd] = false - opts.on('', '--force-ccd float', "") do |param| - $options[:force_ccd] = param.to_f - end - - - $options[:force_focal] = false - opts.on('', '--force-focal float', "") do |param| - $options[:force_focal] = param.to_f - end - - $options[:cmvs_max_images] = 100 - opts.on('', '--cmvs-max-images int', "") do |param| - $options[:cmvs_max_images] = param.to_i - end - - $options[:matcher_ratio] = 0.6 - opts.on('', '--matcher-ratio float', "") do |param| - $options[:matcher_ratio] = param.to_f - end - - $options[:matcher_threshold] = 2.0 - opts.on('', '--matcher-threshold float', "") do |param| - $options[:matcher_threshold] = param.to_f - end - - $options[:pmvs_min_image_num] = 3 - opts.on('', '--pmvs-minImageNum n', "") do |param| - $options[:pmvs_min_image_num] = param.to_i - end - $options[:pmvs_wsize] = 7 - opts.on('', '--pmvs-wsize int', "") do |param| - $options[:pmvs_wsize] = param.to_i - end - $options[:pmvs_threshold] = 0.7 - opts.on('', '--pmvs-threshold float', "") do |param| - $options[:pmvs_threshold] = param.to_f - end - $options[:pmvs_csize] = 2 - opts.on('', '--pmvs-csize int', "") do |param| - $options[:pmvs_csize] = param.to_i - end - $options[:pmvs_level] = 1 - opts.on('', '--pmvs-level int', "") do |param| - $options[:pmvs_level] = param.to_i - end - - opts.on( '-h', '--help', 'Display this screen' ) do - puts opts - exit - end -end - -begin - optparse.parse! - - $options[:path_base] = ARGV[0] if ARGV.length == 1 - $options[:path_base] = File.expand_path($options[:path_base]) - - $options[:path_bin] = File.expand_path(File.dirname(__FILE__)) + "/bin" - - require "#{File.expand_path(File.dirname(__FILE__))}/ccd_defs.rb" - - begin - puts " Configuration:" - puts " bin_path = #{$options[:path_bin]}" - puts " base_path = #{$options[:path_base]}" - puts " " - puts " start_with = #{$options[:start_with]}" - puts " end_with = #{$options[:end_with]}" - puts " " - puts " resize_to = #{$options[:resize_to]}" - puts " " - puts " match_size = #{$options[:match_size]}" - puts " matcher_ratio = #{$options[:matcher_ratio]}" - puts " matcher_threshold = #{$options[:matcher_threshold]}" - puts " " - puts " cmvs_max_images = #{$options[:cmvs_max_images]}" - puts " " - puts " pmvs_threshold = #{$options[:pmvs_threshold]}" - puts " pmvs_csize = #{$options[:pmvs_csize]}" - puts " pmvs_level = #{$options[:pmvs_level]}" - puts " pmvs_min_image_num = #{$options[:pmvs_min_image_num]}" - puts " pmvs_wsize = #{$options[:pmvs_wsize]}" - puts " " - end - - file_objects = [] - - source_files = `ls -1 #{$options[:path_base]} | egrep "\.[jJ]{1}[pP]{1}[eE]{0,1}[gG]{1}"`.split("\n") - file_objects = Parallel.map(source_files) { |source_file| - file_object = Hash.new - - file_object[:file_name] = "#{source_file}" - file_object[:file_basename] = "#{source_file}".sub(/\.[^\.]+$/, "") - file_object[:path_src] = "#{$options[:path_base]}/#{source_file}" - - jhead_text = `jhead #{file_object[:path_src]}` - file_object[:jhead] = Hash.new - - jhead_text.split("\n").each { |jhead_line| - jhead_parts = jhead_line.split(/\ +:\ /) - - file_object[:jhead][jhead_parts[0].to_sym] = jhead_parts[1] if jhead_parts.length == 2 - } - - file_object[:model_id] = "#{file_object[:jhead][:'Camera make']} #{file_object[:jhead][:'Camera model']}" - - file_object[:width],file_object[:height] = [file_object[:jhead][:Resolution].split(" x ")[0].to_i, file_object[:jhead][:Resolution].split(" x ")[1].to_i] if file_object[:jhead][:Resolution] - - - if file_object[:jhead][:'CCD width'] - file_object[:ccd] = file_object[:jhead][:'CCD width'][/([\.0-9]+)/].to_f - end - - if file_object[:jhead][:'Focal length'] - file_object[:focal] = file_object[:jhead][:'Focal length'][/([\.0-9]+)/].to_f - end - - file_object[:ccd] = $ccd_widths[file_object[:model_id].to_sym] unless file_object[:ccd] - - file_object[:focal] = $options[:force_focal] if $options[:force_focal] - file_object[:ccd] = $options[:force_ccd] if $options[:force_ccd] - - if file_object[:focal] && file_object[:ccd] && file_object[:width] && file_object[:height] - file_object[:focal_px] = (file_object[:focal] / file_object[:ccd]) * [[file_object[:width], file_object[:height]].max.to_f, $options[:resize_to].to_f].min - puts "#{file_object[:path_src]} – using image with ccd width: #{file_object[:ccd]} and focal length: #{file_object[:focal]}" - file_object - else - if !file_object[:ccd] - puts "no ccd width found for #{file_object[:model_id]}" - end - end - } - - file_objects = file_objects.select {|file_object| file_object} - - puts "found #{file_objects.length} usable objects" - - def get_feature_count (bin_file_name) - io = File.open(bin_file_name, "rb") - feature_count = io.read(4).unpack("L") - io.close - - feature_count.first.to_i - end - - def match?(i, j, file_object_i, file_object_j, path) - pairwise_match = "#{path}/#{i}-#{j}.txt" - - !File.exists?(pairwise_match) - end - def match (i, j, file_object_i, file_object_j, path, index) - pairwise_match = "#{path}/.#{i}-#{j}.txt" - done_pairwise_match = "#{path}/#{i}-#{j}.txt" - - file_object_i_key = "#{path}/../#{file_object_i[:file_basename]}.key" - file_object_j_key = "#{path}/../#{file_object_j[:file_basename]}.key" - - feature_count_i = get_feature_count("#{path}/../#{file_object_i[:file_basename]}.key.bin") - feature_count_j = get_feature_count("#{path}/../#{file_object_j[:file_basename]}.key.bin") - - `touch '#{pairwise_match}' && '#{$options[:path_bin]}/KeyMatch' '#{file_object_i_key}' '#{file_object_j_key}' '#{pairwise_match}' #{$options[:matcher_ratio]} #{$options[:matcher_threshold]}` unless File.exists?(pairwise_match) - matches =`cat '#{pairwise_match}' | wc -l`.to_i - - if matches > 0 - puts "%6d / %6d - %6.2f%% matches between #{file_object_i[:file_name]}, #{file_object_j[:file_name]}" % [$prog_start + index + 1, $prog_total, matches.to_f*100/([feature_count_i, feature_count_j].min)] - end - - FileUtils.mv(pairwise_match, done_pairwise_match) - end - - def get_keypoints_for_file?(file_object, path, size) - path_matching_base_name = "#{path}/#{file_object[:file_basename]}" - path_matching_key_bin = "#{path_matching_base_name}.key.bin" - path_matching_key_gz = "#{path_matching_base_name}.key.gz" - path_matching_jpg = "#{path_matching_base_name}.jpg" - - !(File.exists?(path_matching_jpg) && File.exists?(path_matching_key_bin) && File.exists?(path_matching_key_gz)) - end - def get_keypoints_for_file (file_object, path, size, index) - path_matching_base_name = "#{path}/.#{file_object[:file_basename]}" - path_matching_jpg = "#{path_matching_base_name}.jpg" - path_matching_pgm = "#{path_matching_base_name}.pgm" - path_matching_sift = "#{path_matching_base_name}.key.sift" - path_matching_key = "#{path_matching_base_name}.key" - path_matching_key_bin = "#{path_matching_base_name}.key.bin" - path_matching_key_gz = "#{path_matching_base_name}.key.gz" - - done_path_matching_base_name = "#{path}/#{file_object[:file_basename]}" - done_path_matching_key_bin = "#{done_path_matching_base_name}.key.bin" - done_path_matching_key_gz = "#{done_path_matching_base_name}.key.gz" - done_path_matching_jpg = "#{done_path_matching_base_name}.jpg" - - - `convert -format jpg -resize #{size}x#{size} -quality 100 '#{file_object[:path_src]}' '#{path_matching_jpg}'` - `convert -format pgm -resize #{size}x#{size} -quality 100 '#{file_object[:path_src]}' '#{path_matching_pgm}'` - - `'#{$options[:path_bin]}/vlsift' '#{path_matching_pgm}' -o '#{path_matching_sift}'` - - `perl '#{$options[:path_bin]}/../convert_vlsift_to_lowesift.pl' '#{path_matching_base_name}'` - - `gzip -f '#{path_matching_key}' && rm -f '#{path_matching_sift}' && rm -f '#{path_matching_pgm}'` - - feature_count = get_feature_count(path_matching_key_bin) - - puts "%6d / %6d - got #{feature_count} keypoints from #{file_object[:file_name]} @ #{size}px" % [$prog_start + index + 1, $prog_total] - - FileUtils.mv(path_matching_key_bin , done_path_matching_key_bin ) - FileUtils.mv(path_matching_key_gz , done_path_matching_key_gz ) - FileUtils.mv(path_matching_jpg , done_path_matching_jpg ) - end - - if file_objects.length > 0 - job_options = Hash.new - - job_options[:path] = "#{$options[:path_base]}/__reconstruction-#{$options[:resize_to]}" - - Dir::mkdir(job_options[:path]) unless File.directory?(job_options[:path]) - - ### MATCHING - - job_options[:path_matching] = "#{$options[:path_base]}/__pre_matching-#{$options[:match_size]}" - job_options[:path_matchinglarge] = "#{job_options[:path]}" - job_options[:path_matching_pairs] = "#{$options[:path_base]}/__pre_matching-#{$options[:match_size]}/_pairs" - job_options[:path_matchinglarge_pairs] = "#{job_options[:path_matchinglarge]}/_pairs" - - Dir::mkdir(job_options[:path_matching]) unless File.directory?(job_options[:path_matching]) - Dir::mkdir(job_options[:path_matchinglarge]) unless File.directory?(job_options[:path_matchinglarge]) - Dir::mkdir(job_options[:path_matching_pairs]) unless File.directory?(job_options[:path_matching_pairs]) - Dir::mkdir(job_options[:path_matchinglarge_pairs]) unless File.directory?(job_options[:path_matchinglarge_pairs]) - - - - puts "\n**\n** GETTING KEYPOINTS SMALL VERSION\n** #{Time.now}\n\n" - - file_objects_todo = file_objects.select { |file_object| get_keypoints_for_file?(file_object, job_options[:path_matching], $options[:match_size]) } - - $prog_start = (file_objects.length - file_objects_todo.length) - $prog_total = file_objects.length - Parallel.each_with_index(file_objects_todo) { |file_object, index| get_keypoints_for_file(file_object, job_options[:path_matching], $options[:match_size], index) } - - puts (file_objects_todo.empty? ? "nothing to do" : "done") - - - - puts "\n**\n** MATCHING SMALL VERSION\n** #{Time.now}\n\n" - - match_indeces = Array.new - - (0...file_objects.length).inject(match_indeces) { |memo, i| - (i+1...file_objects.length).inject(memo) { |memo, j| - memo.push([i, j]) - } - } - - match_indeces_todo = match_indeces.select { |i,j| match?(i, j, file_objects[i], file_objects[j], job_options[:path_matching_pairs]) } - - $prog_start = (match_indeces.length - match_indeces_todo.length) - $prog_total = match_indeces.length - Parallel.each_with_index(match_indeces_todo) { |(i, j), index| - match(i, j, file_objects[i], file_objects[j], job_options[:path_matching_pairs], index) - } - - puts (match_indeces_todo.empty? ? "nothing to do" : "done") - - - - puts "\n**\n** GETTING KEYPOINTS BIG VERSION\n** #{Time.now}\n\n" - - file_objects_todo = file_objects.select { |file_object| get_keypoints_for_file?(file_object, job_options[:path_matchinglarge], $options[:resize_to]) } - - $prog_start = (file_objects.length - file_objects_todo.length) - $prog_total = file_objects.length - Parallel.each_with_index(file_objects_todo, :in_processes => 4) { |file_object, index| get_keypoints_for_file(file_object, job_options[:path_matchinglarge], $options[:resize_to], index) } - - puts (file_objects_todo.empty? ? "nothing to do" : "done") - - - - puts "\n**\n** MATCHING BIG VERSION\n** #{Time.now}\n\n" - - matches_files = `ls -1 #{job_options[:path_matching_pairs]} | egrep "\.txt"`.split("\n").map {|e| (e.sub("\.txt", "").split("-")).map{|n| n.to_i} } - matches_files = matches_files.select {|i,j| File.size?("#{job_options[:path_matching_pairs]}/#{i}-#{j}.txt") } - - matches_files_todo = matches_files.select { |i, j| match?(i, j, file_objects[i], file_objects[j], job_options[:path_matchinglarge_pairs]) } - - $prog_start = (matches_files.length - matches_files_todo.length) - $prog_total = matches_files.length - Parallel.each_with_index(matches_files_todo) { |(i, j), index| match(i, j, file_objects[i], file_objects[j], job_options[:path_matchinglarge_pairs], index) } - - puts (matches_files_todo.empty? ? "nothing to do" : "done") - - - - puts "\n**\n** RUNNING BUNDLER\n** #{Time.now}\n\n" - - job_options[:path_bundle] = "#{job_options[:path]}/bundle" - job_options[:path_pmvs] = "#{job_options[:path]}/pmvs" - job_options[:path_pmvs_txt] = "#{job_options[:path]}/pmvs/txt" - job_options[:path_pmvs_visualize] = "#{job_options[:path]}/pmvs/visualize" - job_options[:path_pmvs_models] = "#{job_options[:path]}/pmvs/models" - - job_options[:file_bundler_filelist] = "#{job_options[:path]}/_bundler_list.txt" - job_options[:file_bundler_options] = "#{job_options[:path]}/_bundler_options.txt" - job_options[:file_bundler_matches_init] = "#{job_options[:path]}/_bundler_matches.init.txt" - - Dir::mkdir(job_options[:path_bundle]) unless File.directory?(job_options[:path_bundle]) - Dir::mkdir(job_options[:path_pmvs]) unless File.directory?(job_options[:path_pmvs]) - Dir::mkdir(job_options[:path_pmvs_txt]) unless File.directory?(job_options[:path_pmvs_txt]) - Dir::mkdir(job_options[:path_pmvs_visualize]) unless File.directory?(job_options[:path_pmvs_visualize]) - Dir::mkdir(job_options[:path_pmvs_models]) unless File.directory?(job_options[:path_pmvs_models]) - -# files_for_bundler = `ls -1 #{job_options[:path_matchinglarge]} | egrep "\.[jJ]{1}[pP]{1}[eE]{0,1}[gG]{1}$"`.split("\n").sort -# files_for_bundler = file_objects.select { |file_object| files_for_bundler.include?(file_object[:file_name]) } - - File.open(job_options[:file_bundler_filelist], 'w') do |file| - file_objects.each { |file_object| - file.puts "./%s 0 %0.5f" % ["#{file_object[:file_basename]}.jpg", file_object[:focal_px]] - } - end - - File.open(job_options[:file_bundler_options], 'w') do |file| - file.puts "--match_table _bundler_matches.init.txt" - file.puts "--output bundle.out" - file.puts "--output_all bundle_" - file.puts "--output_dir bundle" - file.puts "--variable_focal_length" - file.puts "--use_focal_estimate" - file.puts "--constrain_focal" - file.puts "--constrain_focal_weight 0.01" - file.puts "--estimate_distortion" - file.puts "--run_bundle" - end - - File.open(job_options[:file_bundler_matches_init], 'w') do |file| - matches_files = `ls -1 #{job_options[:path_matchinglarge_pairs]} | egrep "\.txt"`.split("\n").map {|e| (e.sub("\.txt", "").split("-")).map{|n| n.to_i} } - matches_files = matches_files.sort - - matches_files.each {|i,j| - if File.size?("#{job_options[:path_matchinglarge_pairs]}/#{i}-#{j}.txt") - file.puts "#{i} #{j}" - file.puts File.read("#{job_options[:path_matchinglarge_pairs]}/#{i}-#{j}.txt") - end - } - end - - Dir.chdir("#{job_options[:path]}") - - system("'#{$options[:path_bin]}/bundler' '#{job_options[:file_bundler_filelist]}' --options_file '#{job_options[:file_bundler_options]}'") - - puts "\n**\n** RUNNING BUNDLE2PMVS\n** #{Time.now}\n\n" - system("'#{$options[:path_bin]}/Bundle2PMVS' '#{job_options[:file_bundler_filelist]}' 'bundle/bundle.out'") - - puts "\n**\n** RUNNING RADIALUNDISTORT\n** #{Time.now}\n\n" - system("'#{$options[:path_bin]}/RadialUndistort' '#{job_options[:file_bundler_filelist]}' 'bundle/bundle.out' 'pmvs'") - - i = 0 - - file_objects.each { |file_object| - if File.exist?("#{job_options[:path]}/pmvs/#{file_object[:file_basename]}.rd.jpg") - nr = "%08d" % [i] - - puts "#{job_options[:path]}/pmvs/#{file_object[:file_basename]}.rd.jpg", "#{job_options[:path]}/pmvs/visualize/#{nr}.jpg" - - FileUtils.mv("#{job_options[:path]}/pmvs/#{file_object[:file_basename]}.rd.jpg", "#{job_options[:path]}/pmvs/visualize/#{nr}.jpg") - FileUtils.mv("#{job_options[:path]}/pmvs/#{nr}.txt", "#{job_options[:path]}/pmvs/txt/#{nr}.txt") - - i += 1 - end - } - - puts "\n**\n** RUNNING CMVS\n** #{Time.now}\n\n" - system("'#{$options[:path_bin]}/cmvs' pmvs/ #{$options[:cmvs_max_images]} #{Parallel.processor_count}"); - - puts "\n**\n** GENOPTION CMVS\n** #{Time.now}\n\n" - system("'#{$options[:path_bin]}/genOption' pmvs/ #{$options[:pmvs_level]} #{$options[:pmvs_csize]} #{$options[:pmvs_threshold]} #{$options[:pmvs_wsize]} #{$options[:pmvs_min_image_num]} #{Parallel.processor_count}"); - - puts "\n**\n** GENOPTION PMVS\n** #{Time.now}\n\n" - system("'#{$options[:path_bin]}/pmvs2' pmvs/ option-0000"); - - end -end