Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
  • v0.0.1
  • v0.0.0
3 results
Blame Permalink
run_oracle_wrapper_local.py 2.27 KiB
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 






# This is the the file to be run on the oracle cloud

import oci
import argparse
import os
import io
import shutil
import yaml

from run import main

def run_oracle_wrapper(path_to_config_file):

    # read the config file from config folder
    with open(path_to_config_file) as f:
        config_flow_params = yaml.load(f, Loader=yaml.FullLoader)

    # read system environment variables
    # input_location = os.environ['OBJ_INPUT_LOCATION']
    # output_location = os.environ['OBJ_OUTPUT_LOCATION']

    # remap the input and output locations
    # input_location = input_location.replace("@axqlz2potslu", "").replace("oci://", "/mnt/")
    # output_location = output_location.replace("@axqlz2potslu", "").replace("oci://", "/mnt/")

    input_location = "local_folder_in/las_files/"
    output_location = "local_folder_out/las_files/"

    # copy files from input_location to the input folder
    shutil.copytree(input_location, config_flow_params['general']['input_folder'])

    # run the main function
    main(path_to_config_file)

    # instance segmentation is set to true
    if config_flow_params['general']['run_instance_segmentation']:
        path_to_the_output_folder = os.path.join(config_flow_params['general']['output_folder'], 'instance_segmented_point_clouds_with_ground')
    else:
        path_to_the_output_folder = config_flow_params['general']['output_folder']

    # zip the files in path_to_the_output_folder
    zip_file_name  = 'results'
    shutil.make_archive(zip_file_name, 'zip', path_to_the_output_folder) # this will be done in the current folder
    shutil.copy('results.zip', path_to_the_output_folder)

    # copy the zip file and other files to the output location
    for filename in os.listdir(path_to_the_output_folder):
        src_file = os.path.join(path_to_the_output_folder, filename)
        dst_file = os.path.join(output_location, filename)
        shutil.copy(src_file, dst_file)

if __name__ == '__main__':
    # use argparse to get the path to the config file
    parser = argparse.ArgumentParser()
    parser.add_argument("--path_to_config_file", type=str, default="./config/config.yaml")
    args = parser.parse_args()

    # run the main function
    print('Running the main function in run_oracle_wrapper.py')
    run_oracle_wrapper(args.path_to_config_file)