Python: copier des images RVB à l'intérieur de la fonction

Question

Je suis nouveau sur Python. J'ai écrit (Python 3.6.1) une fonction simple pour la rotation des images à l'aide des outils skimage.transform:

def rotate_20rand(X): 
    X_aug = np.zeros_like(X) 
    for i,img in enumerate(X): 
     X_aug[i] = transform.rotate(img, angle=np.random.uniform(-20, 20)) 
    return X_aug 

Je passe mon jeu de données np.ndarray de (34799, 32, 32, 3) images à elle. Si les images sont en niveaux de gris, la sortie de la fonction est correcte. Cependant, après le passage des images RVB X_aug contient tous les zéros.

---

J'ai essayé ce qui suit:

• D'abord, je vérifie si la fonction transform.rotate(...) fonctionne après que je passe l'image RVB à elle. Puis j'ai pensé que le problème est: X_aug[i] = transform.rotate(img, angle=np.random.uniform(-20, 20)) et essayé différentes méthodes copy.

Aucun de ceux-ci n'a aidé et je ne sais pas ce que je fais mal.

Answer 1

Votre code fonctionne pour moi. Il produit des valeurs non nulles dans X_aug pour tout type de tableau d'entrée.

Essayez ceci:

def rotate_20rand(X): 
    return np.stack((transform.rotate(img, angle=np.random.uniform(-20, 20)) for img in X))