README.md

metadata

tags:
  - object-detection
  - vision
  - fasterrcnn
  - onnx
license: apache-2.0

Custom Faster R-CNN Model

Usage

PyTorch

from inference import FasterRCNNInference
from PIL import Image

model = FasterRCNNInference(".")
image = Image.open("your_image.jpg")
results = model.predict([image])