چطور داده‌ها را برای مدل آماده کنیم؟

پس از جمع‌آوری داده باید آن‌ها را نرمال‌سازی و به فرمت مناسب تبدیل کنید. برای تصاویر، اندازه یکسان و مقدار پیکسل‌ها بین ۰ تا ۱ ضروری است. در PyTorch می‌توان از Dataset و DataLoader برای مدیریت داده استفاده کرد. تقسیم داده به مجموعه‌ی آموزش و تست به ارزیابی مدل کمک می‌کند. این مرحله پایه‌ی موفقیت در آموزش مدل است.