ObjectDetection / app.py
IS361Group4's picture
สร้าง app.py
6dc3acb verified
raw
history blame
393 Bytes
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
# Load model directly
from transformers import AutoImageProcessor, AutoModelForObjectDetection
processor = AutoImageProcessor.from_pretrained("facebook/detr-resnet-50")
model = AutoModelForObjectDetection.from_pretrained("facebook/detr-resnet-50")