thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix import

Browse Source
This commit is contained in:
Anvar Nazar
2023-09-26 17:04:52 +05:30
committed by GitHub
parent b738644f6e
commit a9d2f13544
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils/export_yoloV8.py
View File

@@ -7,7 +7,7 @@ import torch
import torch.nn as nn import torch.nn as nn
from copy import deepcopy from copy import deepcopy
from ultralytics import YOLO from ultralytics import YOLO
from ultralytics.yolo.utils.torch_utils import select_device from ultralytics.utils.torch_utils import select_device
from ultralytics.nn.modules import C2f, Detect, RTDETRDecoder from ultralytics.nn.modules import C2f, Detect, RTDETRDecoder