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
|
import torch.nn as nn
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
CHECK_PYTORCH_18 = torch.__version__.startswith("1.8")
|
CHECK_PYTORCH_18 = torch.__version__.startswith("1.8")
|
||||||
|
|
||||||
# Import 'ultralytics' package or install if missing
|
# Import 'ultralytics' package or install if missing
|
||||||
try:
|
try:
|
||||||
import ultralytics
|
import ultralytics
|
||||||
|
Loading…
x
Reference in New Issue
Block a user