Scipy drehen und Zoomen eines Bildes ohne änderung der Abmessungen

Für mein neuronales Netz ich möchte ergänzen meine Ausbildung Daten, indem Sie kleine zufällige Rotationen und zooms zu meinen Bildern. Das Problem, das ich habe ist, dass scipy ist die änderung der Größe meiner Bilder, wenn es gilt, die Drehungen und zooms. Ich muss, um einfach befestigen Sie die Kanten aus, wenn ein Teil des Bildes geht out of bounds. Alle meine Bilder müssen die gleiche Größe haben.

def loadImageData(img, distort = False):
    c, fn = img
    img = scipy.ndimage.imread(fn, True)

    if distort:
        img = scipy.ndimage.zoom(img, 1 + 0.05 * rnd(), mode = 'constant')
        img = scipy.ndimage.rotate(img, 10 * rnd(), mode = 'constant')
        print(img.shape)

    img = img - np.min(img)
    img = img / np.max(img)
    img = np.reshape(img, (1, *img.shape))

    y = np.zeros(ncats)
    y[c] = 1
    return (img, y)

InformationsquelleAutor chasep255 | 2016-05-09

Schreibe einen Kommentar