trashscan8n

YOLOv8 mobile_model_package_yolov8n_v2 ONNX model for waste classification.

Classes

Index	Class
0	paper
1	plastic
2	metal
3	organic
4	other

Usage

import onnxruntime
import numpy as np
from PIL import Image

session = onnxruntime.InferenceSession("best.onnx")
input_name = session.get_inputs()[0].name
# Input: [1, 3, 640, 640] normalized float32
# Output: [1, 9, 8400] = [cx, cy, w, h, p0, p1, p2, p3, p4]

Model Specs

• Input size: 640x640
• Format: ONNX
• Confidence threshold: 0.25
• IoU threshold: 0.5