martinsbruveris/tensorflow-image-models
این پروژه یک نسخهٔ تنسورفلو از کتابخانهٔ معروف PyTorch Image Models (timm) است که شامل مدل‌های یادگیری عمیق برای پردازش تصویر با وزن‌های از پیش آموزش دیده است. ️
کاربرد:
این کتابخانه به تو اجازه می‌ده تا به راحتی از مدل‌های پیشرفته تشخیص تصویر مثل ResNet, EfficientNet, ViT و ده‌ها مدل دیگر در محیط تنسورفلو استفاده کنی، بدون اینکه نیاز باشه از صفر کد بنویسی. همچنین وزن‌های از پیش آموزش دیده رو هم داره، پس می‌تونی مستقیماً شروع به fine-tuning یا استنتاج کنی.
در چه شرایطی بهتره استفاده شود؟
اگر داری روی یک پروژهٔ تشخیص تصویر، طبقه‌بندی، استخراج ویژگی یا انتقال یادگیری (transfer learning) کار می‌کنی و ترجیح می‌دی از تنسورفلو استفاده کنی، این کتابخانه گزینهٔ عالی‌ایه. به‌ویژه وقتی می‌خوای از مدل‌های پیچیده و به‌روز بدون دردسر استفاده کنی.
چند مثال از موارد استفاده:
طبقه‌بندی نژاد سگ از روی عکس با استفاده از یک EfficientNet آماده
تشخیص بیماری در برگ گیاهان با fine-tuning یک ConvNeXt
استخراج ویژگی از تصاویر برای سیستم جستجوی تصویری با ResNet50
مقایسه عملکرد مدل‌های مختلف در یک دیتاست کوچک با استفاده از وزن‌های pretrained
- ##tensorflow
- ##deeplearning
- ##image
- ##transfer
- ##computer
- ##timm
- ##pretrained
- ##ViT
- ##EfficientNet
- ##ResNet