افزودن تابع دلخواه به image data generator

Question

من در پردازش تصویر از image data generator کراس استفاده میکنم. از shift  و flip . الان لازم دارم یک تابع دیگه که خودم تعریف کردم رو به این ۲ تا اضافه کنم ولی بر روی همه تصاویر اعمال نشه. یعنی یا flip  یا shift یا تابع من روی تصویر اعمال بشه . پس از preprocessing function نمیشه استفاده کرد! روشی سراغ دارین؟

Answer 1

کلاس ImageDataGenerator پارامتری داره که preprocessing_function را از شما می گیره برای حل مسٔله شما می تونید تابع preprocessing خودتون را تعریف کنید و تو آن با توجه به شرایط تصمیم بگیرید.

 

def my_preprocessor(image):
    # perform augmentations here



image_generator = ImageDataGenerator(..., preprocessing_function=my_preprocessor)

 

Comments

اگه به متن سوالم خوب دقت کنید نوشتم که نمیشه از preprocessing function استفاده کنم. چون اون تابع روی تمام داده ها اعمال میشه. من میخوام اگه تابع من اجرا شد دیگه flip  و shift اعمال نشه رو تصویرم.

---

بله من هم با توجه به متن سوال پاسخ دادم شما تو تابع my_preprocessor می تونید تصمیم بگیرید یا flip  یا shift یا تابع شما روی تصویر اعمال بشه مثلا یک متغیر تصادفی در تولید کنید که نشان دهنده نوع عملیات پیش پردازش باشه تو این مثال بین ۰ تا ۲ مثلا ۰ برای flip و ۱ برای shift و ۲ برای تابع دلخواه شما . با دیدن هر حالت تابع موردنظر را اعمال کنید.

---

اهان بله این یه روش خوبه. مرسی