efee1712aa
* fix.. speed_limit error... * draw tpms settings. * fix.. traffic light stopping only.. * fix.. waze cam * fix.. waze... * add setting (Enable comma connect ) * auto detect LFA2 * fix.. cruisespeed1 * vff2 driving model. * fix.. * agnos 12.3 * fix.. * ff * ff * test * ff * fix.. drawTurnInfo.. * Update drive_helpers.py * fix.. support eng voice eng sounds fix settings... english fix.. mph.. fix.. roadlimit speed bug.. * new vff model.. 250608 * fix soundd.. * fix safe exit speed.. * fix.. sounds. * fix.. radar timeStep.. * KerryGold model * Update drive_helpers.py * fix.. model. * fix.. * fix.. * Revert "fix.." This reverts commit b09ec459afb855c533d47fd7e8a1a6b1a09466e7. * Revert "fix.." This reverts commit 290bec6b83a4554ca232d531a911edccf94a2156. * fix esim * add more acc table. 10kph * kg update.. * fix cruisebutton mode3 * test atc..cond. * fix.. canfd * fix.. angle control limit
20 lines
861 B
Python
20 lines
861 B
Python
from pathlib import Path
|
|
from examples.yolov8 import YOLOv8, get_weights_location
|
|
from tinygrad.tensor import Tensor
|
|
from tinygrad.nn.state import safe_save
|
|
from extra.export_model import export_model
|
|
from tinygrad.device import Device
|
|
from tinygrad.nn.state import safe_load, load_state_dict
|
|
|
|
if __name__ == "__main__":
|
|
Device.DEFAULT = "WEBGPU"
|
|
yolo_variant = 'n'
|
|
yolo_infer = YOLOv8(w=0.25, r=2.0, d=0.33, num_classes=80)
|
|
state_dict = safe_load(get_weights_location(yolo_variant))
|
|
load_state_dict(yolo_infer, state_dict)
|
|
prg, inp_sizes, out_sizes, state = export_model(yolo_infer, Device.DEFAULT.lower(), Tensor.randn(1,3,640,640), model_name="yolov8")
|
|
dirname = Path(__file__).parent
|
|
safe_save(state, (dirname / "net.safetensors").as_posix())
|
|
with open(dirname / f"net.js", "w") as text_file:
|
|
text_file.write(prg)
|