Comment faire une rotation d'image dans tensorflow?

Question

Je veux faire de sérieuses augmentations d'image sur mon propre jeu de données car il est vraiment petit. Tensorflow fournit la bibliothèque d'images pour faire toute l'augmentation, ce qui est très pratique. Mais il n'y a pas de fonctionnalité de "rotation". Je sais qu'il y a un problème suivi here, mais il est si long et semble compliqué. Y a-t-il un moyen simple de faire la rotation de l'image en tensorflow? Merci beaucoup.

Answer 1

Si vous souhaitez utiliser votre propre rotation d'image, vous pouvez probablement le faire avec py_func, PIL et votre propre processeur de file d'attente. Alternativement, vous pouvez écrire votre propre file d'attente pour nourrir des exemples. Une autre façon laide, est atomiquement (écrire + renommer) tourner ces images et les écrire sur le disque avec le même nom de fichier, tandis que l'entraînement se poursuit.

Quoi que vous fassiez, rappelez-vous que nous nous entraînons en lots, alors assurez-vous de servir les rotations un à la fois, toutes les variantes de la même image (shuffle(rotations([image]) est jointive pas rotations(shuffle([image]))), sinon la formation se conduisent mal.