تغییرات در کد deep learning موجود

با سلام. مهندس جان من چند وقت پیش یکسری زحمت به شما دادم خاطرتون باشه در مورد مدل دیپ برای تشخیص سرطان از تصاویر مامو بود. الان کدها آماده شده و اوکی شده همه چی. حالا بعد این همه وقت استاد راهنما فرمودند که شما قبل از دیتا اسپلیت اومدین آگمنت کردین و این اشتباه هست. حالا کد باید اصلاح بشه. اصلاح مورد نظر هم این هست که الان نمونه تصاویر لود میشه ، بعد آگمنت میشه و بعد اسپلیت میشه. الان باید تصاویر قبل آگمنت جداسازی بشن و بعد هر دسته جداگانه آگمنت بشن. میخواستم اگر امکانش باشه یه برآورد هزینه و زمان بدین اگر امکان داره زحمت بکشید. با سپاس از شما.