Food Vision

تشخیص غذا با مدل EfficientNetB2 و مجموعه‌داده‌ی Food-101

زمینه‌های مرتبط: یادگیری عمیق، یادگیری ماشین، برنامه‌نویسی

معرفی پروژه

وقتی با دوره‌ی «PyTorch for Deep Learning Bootcamp» در Udemy (۵۲ ساعت و ۱۴ دقیقه) وارد دنیای یادگیری عمیق شدم، آخرش یک مدل خیلی جالب از کار درآمد. فکرش را بکنید: مدلی که فقط با یک نگاه به عکس بفهمد چه غذایی جلویش است! 🍔📸 کار از همین‌جا شروع شد: ساختن شبکه‌ای عصبی که هر غذایی را از روی عکسش بشناسد. بیایید مزه‌ی پنهان‌شده در پیکسل‌ها را با هم ببینیم.

مراحل کار

  1. دانلود مجموعه‌داده‌ی Food101 از این لینک.
  2. آماده کردن داده‌ها، همراه با transformها و augmentationها.
  3. آماده کردن مدل EfficientNetB2 با یادگیری انتقالی.
  4. آموزش مدل و گذاشتنش روی Hugging Face.

کد و منابع

این مدل با PyTorch ساخته شده است.

همه‌ی کدها و نتیجه‌ها این‌جاست:

Github-Food-Vision
HuggingFace-Space-Food-Vision

برگشت به بالای صفحه