├── .gitignore ├── LICENSE ├── README.md ├── coco_classes.txt ├── sahi ├── __init__.py ├── annotation.py ├── auto_model.py ├── cli.py ├── models │ ├── __init__.py │ ├── base.py │ ├── detectron2.py │ ├── huggingface.py │ ├── mmdet.py │ ├── torchvision.py │ ├── yolov5.py │ └── yolox_onnx.py ├── postprocess │ ├── __init__.py │ ├── combine.py │ ├── legacy │ │ ├── __init__.py │ │ └── combine.py │ └── utils.py ├── predict.py ├── prediction.py ├── scripts │ ├── __init__.py │ ├── coco2fiftyone.py │ ├── coco2yolov5.py │ ├── coco_error_analysis.py │ ├── coco_evaluation.py │ ├── predict.py │ ├── predict_fiftyone.py │ └── slice_coco.py ├── slicing.py └── utils │ ├── __init__.py │ ├── coco.py │ ├── compatibility.py │ ├── cv.py │ ├── detectron2.py │ ├── fiftyone.py │ ├── file.py │ ├── huggingface.py │ ├── import_utils.py │ ├── mmdet.py │ ├── shapely.py │ ├── torch.py │ ├── torchvision.py │ ├── versions.py │ └── yolov5.py ├── sample_prediction.py ├── sample_sliced_prediction.py ├── test.mp4 └── yolox ├── config.json ├── model └── yolox_nano.onnx └── yolox_onnx.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/README.md -------------------------------------------------------------------------------- /coco_classes.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/coco_classes.txt -------------------------------------------------------------------------------- /sahi/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/__init__.py -------------------------------------------------------------------------------- /sahi/annotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/annotation.py -------------------------------------------------------------------------------- /sahi/auto_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/auto_model.py -------------------------------------------------------------------------------- /sahi/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/cli.py -------------------------------------------------------------------------------- /sahi/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/models/__init__.py -------------------------------------------------------------------------------- /sahi/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/models/base.py -------------------------------------------------------------------------------- /sahi/models/detectron2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/models/detectron2.py -------------------------------------------------------------------------------- /sahi/models/huggingface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/models/huggingface.py -------------------------------------------------------------------------------- /sahi/models/mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/models/mmdet.py -------------------------------------------------------------------------------- /sahi/models/torchvision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/models/torchvision.py -------------------------------------------------------------------------------- /sahi/models/yolov5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/models/yolov5.py -------------------------------------------------------------------------------- /sahi/models/yolox_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/models/yolox_onnx.py -------------------------------------------------------------------------------- /sahi/postprocess/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sahi/postprocess/combine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/postprocess/combine.py -------------------------------------------------------------------------------- /sahi/postprocess/legacy/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /sahi/postprocess/legacy/combine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/postprocess/legacy/combine.py -------------------------------------------------------------------------------- /sahi/postprocess/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/postprocess/utils.py -------------------------------------------------------------------------------- /sahi/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/predict.py -------------------------------------------------------------------------------- /sahi/prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/prediction.py -------------------------------------------------------------------------------- /sahi/scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sahi/scripts/coco2fiftyone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/scripts/coco2fiftyone.py -------------------------------------------------------------------------------- /sahi/scripts/coco2yolov5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/scripts/coco2yolov5.py -------------------------------------------------------------------------------- /sahi/scripts/coco_error_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/scripts/coco_error_analysis.py -------------------------------------------------------------------------------- /sahi/scripts/coco_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/scripts/coco_evaluation.py -------------------------------------------------------------------------------- /sahi/scripts/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/scripts/predict.py -------------------------------------------------------------------------------- /sahi/scripts/predict_fiftyone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/scripts/predict_fiftyone.py -------------------------------------------------------------------------------- /sahi/scripts/slice_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/scripts/slice_coco.py -------------------------------------------------------------------------------- /sahi/slicing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/slicing.py -------------------------------------------------------------------------------- /sahi/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sahi/utils/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/coco.py -------------------------------------------------------------------------------- /sahi/utils/compatibility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/compatibility.py -------------------------------------------------------------------------------- /sahi/utils/cv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/cv.py -------------------------------------------------------------------------------- /sahi/utils/detectron2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/detectron2.py -------------------------------------------------------------------------------- /sahi/utils/fiftyone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/fiftyone.py -------------------------------------------------------------------------------- /sahi/utils/file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/file.py -------------------------------------------------------------------------------- /sahi/utils/huggingface.py: -------------------------------------------------------------------------------- 1 | class HuggingfaceTestConstants: 2 | YOLOS_TINY_MODEL_PATH = "hustvl/yolos-tiny" 3 | -------------------------------------------------------------------------------- /sahi/utils/import_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/import_utils.py -------------------------------------------------------------------------------- /sahi/utils/mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/mmdet.py -------------------------------------------------------------------------------- /sahi/utils/shapely.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/shapely.py -------------------------------------------------------------------------------- /sahi/utils/torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/torch.py -------------------------------------------------------------------------------- /sahi/utils/torchvision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/torchvision.py -------------------------------------------------------------------------------- /sahi/utils/versions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/versions.py -------------------------------------------------------------------------------- /sahi/utils/yolov5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sahi/utils/yolov5.py -------------------------------------------------------------------------------- /sample_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sample_prediction.py -------------------------------------------------------------------------------- /sample_sliced_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/sample_sliced_prediction.py -------------------------------------------------------------------------------- /test.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/test.mp4 -------------------------------------------------------------------------------- /yolox/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/yolox/config.json -------------------------------------------------------------------------------- /yolox/model/yolox_nano.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/yolox/model/yolox_nano.onnx -------------------------------------------------------------------------------- /yolox/yolox_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kazuhito00/sahi-yolox-onnx-sample/HEAD/yolox/yolox_onnx.py --------------------------------------------------------------------------------