Add TorchScript Docs Integrations Page (#8501)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
f8e681c2be
commit
1db8e912a3
9 changed files with 146 additions and 6 deletions
|
|
@ -48,6 +48,7 @@ TensorFlow.js:
|
|||
$ ln -s ../../yolov5/yolov8n_web_model public/yolov8n_web_model
|
||||
$ npm start
|
||||
"""
|
||||
|
||||
import json
|
||||
import os
|
||||
import shutil
|
||||
|
|
@ -1079,9 +1080,9 @@ class Exporter:
|
|||
model = ct.models.MLModel(pipeline.spec, weights_dir=weights_dir)
|
||||
model.input_description["image"] = "Input image"
|
||||
model.input_description["iouThreshold"] = f"(optional) IOU threshold override (default: {nms.iouThreshold})"
|
||||
model.input_description[
|
||||
"confidenceThreshold"
|
||||
] = f"(optional) Confidence threshold override (default: {nms.confidenceThreshold})"
|
||||
model.input_description["confidenceThreshold"] = (
|
||||
f"(optional) Confidence threshold override (default: {nms.confidenceThreshold})"
|
||||
)
|
||||
model.output_description["confidence"] = 'Boxes × Class confidence (see user-defined metadata "classes")'
|
||||
model.output_description["coordinates"] = "Boxes × [x, y, width, height] (relative to image size)"
|
||||
LOGGER.info(f"{prefix} pipeline success")
|
||||
|
|
|
|||
|
|
@ -27,6 +27,7 @@ Usage - formats:
|
|||
yolov8n_edgetpu.tflite # TensorFlow Edge TPU
|
||||
yolov8n_paddle_model # PaddlePaddle
|
||||
"""
|
||||
|
||||
import platform
|
||||
import threading
|
||||
from pathlib import Path
|
||||
|
|
|
|||
|
|
@ -16,6 +16,7 @@ Example:
|
|||
model.tune(data='coco8.yaml', epochs=10, iterations=300, optimizer='AdamW', plots=False, save=False, val=False)
|
||||
```
|
||||
"""
|
||||
|
||||
import random
|
||||
import shutil
|
||||
import subprocess
|
||||
|
|
|
|||
|
|
@ -18,6 +18,7 @@ Usage - formats:
|
|||
yolov8n_edgetpu.tflite # TensorFlow Edge TPU
|
||||
yolov8n_paddle_model # PaddlePaddle
|
||||
"""
|
||||
|
||||
import json
|
||||
import time
|
||||
from pathlib import Path
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue