image_groups.txt file support

Dockerfile

@@ -69,7 +69,7 @@ RUN pip install -U \
   numpy==1.15.4 \
   psutil \
   pyproj \
-  PyYAML \
+  PyYAML==3.13 \
   repoze.lru \
   scipy \
   shapely \

configure.sh

@@ -71,7 +71,7 @@ install() {
                          libboost-thread-dev \
                          python-pyproj
 
-    pip install -U PyYAML \
+    pip install -U PyYAML==3.13 \
                         exifread \
                         gpxpy \
                         xmltodict \

opendm/osfm.py

@@ -57,9 +57,10 @@ def setup(args, images_path, opensfm_path, photos, gcp_path=None, append_config
     if not io.dir_exists(opensfm_path):
         system.mkdir_p(opensfm_path)
 
-    # create file list
     list_path = io.join_paths(opensfm_path, 'image_list.txt')
     if not io.file_exists(list_path) or rerun:
+        
+        # create file list
         has_alt = True
         with open(list_path, 'w') as fout:
             for photo in photos:
@@ -111,9 +112,16 @@ def setup(args, images_path, opensfm_path, photos, gcp_path=None, append_config
         config_filename = io.join_paths(opensfm_path, 'config.yaml')
         with open(config_filename, 'w') as fout:
             fout.write("\n".join(config))
+
+        # check for image_groups.txt (split-merge)
+        image_groups_file = os.path.join(args.project_path, "image_groups.txt")
+        if io.file_exists(image_groups_file):
+            log.ODM_DEBUG("Copied image_groups.txt to OpenSfM directory")
+            io.copy(image_groups_file, os.path.join(opensfm_path, "image_groups.txt"))
     else:
         log.ODM_WARNING("%s already exists, not rerunning OpenSfM setup" % list_path)
 
+
 def feature_matching(opensfm_project_path, rerun=False):
     if not feature_matching_done(opensfm_project_path) or rerun:
         run('extract_metadata', opensfm_project_path)

portable.Dockerfile

@@ -69,7 +69,7 @@ RUN pip install -U \
   numpy==1.15.4 \
   psutil \
   pyproj \
-  PyYAML \
+  PyYAML==3.13 \
   repoze.lru \
   scipy \
   shapely \

scripts/odm_app.py

@@ -65,7 +65,7 @@ class ODMApp:
                                             build_overviews=args.build_overviews,
                                             max_concurrency=args.max_concurrency,
                                             verbose=args.verbose)
-        
+
         if not args.video:
             # Normal pipeline
             self.first_stage = dataset

scripts/splitmerge.py

@@ -1,4 +1,5 @@
-import os, sys
+import os
+import shutil
 from opendm import log
 from opendm import osfm
 from opendm import types
@@ -43,7 +44,7 @@ class ODMSplitStage(types.ODM_Stage):
                     osfm.run("create_submodels", tree.opensfm)
                 else:
                     log.ODM_WARNING("Submodels directory already exist at: %s" % tree.submodels_path)
-                
+
                 # TODO: on a network workflow we probably stop here
                 # and let NodeODM take over
                 # exit(0)