MOJAB 24

کد پایتون برای اجرای مدل yolo11 در gpu

برای اجرای مدل YOLOv11 در GPU با استفاده از کتابخانه PyTorch، می‌توانید از کد زیر استفاده کنید:

import torch
from models import YOLOv11

# ایجاد یک نمونه از مدل YOLOv11
model = YOLOv11()

# انتقال مدل به دستگاه GPU اگر ممکن است
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model.to(device)

# تعریف یک نمونه از داده‌ها
input_data = torch.randn(1, 3, 416, 416).to(device)

# اجرای مدل بر روی داده‌ها
output = model(input_data)

print(output)

در این کد، ابتدا یک نمونه از مدل YOLOv11 ایجاد شده و سپس به دستگاه GPU انتقال داده می‌شود. سپس یک نمونه از داده‌ها تعریف شده و به دستگاه GPU منتقل می‌شود. در نهایت، مدل بر روی داده‌ها اجرا می‌شود و خروجی مدل چاپ می‌شود.