wrapper.py

import sys
from subprocess import call
from cytomine.models import Job
from biaflows import CLASS_OBJTRK
from biaflows.helpers import get_discipline, BiaflowsJob, prepare_data, upload_data, upload_metrics

def main(argv):
    # 0. Initialize Cytomine client and job if necessary and parse inputs
    with BiaflowsJob.from_cli(argv) as nj:
        problem_cls = get_discipline(nj, default=CLASS_OBJTRK)
        is_2d = False
        nj.job.update(status=Job.RUNNING, progress=0,
                      statusComment="Running workflow for problem class '{}'".format(problem_cls))

        # 1. Create working directories on the machine
        # 2. Download the images
        nj.job.update(progress=0, statusComment="Initialisation...")
        in_images, gt_images, in_path, gt_path, out_path, tmp_path = prepare_data(problem_cls, nj, **nj.flags)

        # 3. Call the image analysis workflow using the run script
        nj.job.update(progress=25, statusComment="Launching workflow...")
        command = "octave-cli " \
                  "-qf /app/Track_script.m {} {} {} {} {}" \
                    .format(in_path, out_path, nj.parameters.gblurrad, nj.parameters.meanbox, nj.parameters.offset)					
        return_code = call(command, shell=True, cwd="/app")  # waits for the subprocess to return

        if return_code != 0:
            err_desc = "Failed to execute the Octave script (return code: {})".format(return_code)
            nj.job.update(progress=50, statusComment=err_desc)
            raise ValueError(err_desc)

        # 4. Upload the annotation and labels to Cytomine
        upload_data(problem_cls, nj, in_images, out_path, **nj.flags, is_2d=is_2d, monitor_params={
            "start": 60, "end": 90, "period": 0.1
        })

        # 5. Compute and upload the metrics
        nj.job.update(progress=90, statusComment="Computing and uploading metrics...")
        upload_metrics(problem_cls, nj, in_images, gt_path, out_path, tmp_path, **nj.flags)

        # 6. End
        nj.job.update(status=Job.TERMINATED, progress=100, statusComment="Finished.")


if __name__ == "__main__":
    main(sys.argv[1:])