Spaces:
Runtime error
Runtime error
File size: 498 Bytes
0b7b08a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
"""
Usage example:
import torch
model = torch.hub.load("Megvii-BaseDetection/YOLOX", "yolox_s")
model = torch.hub.load("Megvii-BaseDetection/YOLOX", "yolox_custom",
exp_path="exp.py", ckpt_path="ckpt.pth")
"""
dependencies = ["torch"]
from yolox.models import ( # isort:skip # noqa: F401, E402
yolox_tiny,
yolox_nano,
yolox_s,
yolox_m,
yolox_l,
yolox_x,
yolov3,
yolox_custom
)
|