Exemplo n.º 1
0
            cut['TargetDistances'] = [(0., 0.), (1.5, 0.), (-3., 0.),
                                      (-1.5, 0.), (0, 1.5), (0, -1.5)]
        else:
            cut['TargetDistances'] = args.custom_landmarks
        # number of outputs to be extracted from the image
        # (x and y head coordinates = 2 targets)
    elif cut['Mode'] == 'end-to-end':
        # list of distances to move points on the main axis
        # (this could be made to vectors, if the rotation of the object itself
        # was to be considered (future work)
        # screw: [0, 1.5, 3, -1.5]
        cut['TargetDistances'] = [(0., 0.)]

    # number of outputs to be extracted from the image
    # (x and y head coordinates = 2 targets) plus angle, tilt and resolution
    target_number = cut.target_number()

    # cropped image dimensions
    imageDimensions = (cut['Large']['Width'], cut['Large']['Height'])

    # number of hdf5 samples per drr
    numCuts = args.num_samples

    # if len(strWidth) > 0: widthFinal = int(strWidth)
    # if len(strHeight) > 0: heightFinal = int(strHeight)

    cut.verbose = args.verbose
    cut.show_images = args.plot

    if args.mode == 'range':
        endIndex = args.stop