mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
refactored the code
This commit is contained in:
parent
6695bac0cd
commit
8b9d980dff
@ -21,6 +21,7 @@ import torch
|
||||
import torch.nn as nn
|
||||
from PIL import Image
|
||||
CHECK_PYTORCH_18 = torch.__version__.startswith("1.8")
|
||||
|
||||
# Import 'ultralytics' package or install if missing
|
||||
try:
|
||||
import ultralytics
|
||||
|
Loading…
x
Reference in New Issue
Block a user