basilshaji
commited on
Update yolov5/utils/augmentations.py
Browse files
yolov5/utils/augmentations.py
CHANGED
@@ -10,8 +10,8 @@ import torch
|
|
10 |
import torchvision.transforms as T
|
11 |
import torchvision.transforms.functional as TF
|
12 |
|
13 |
-
from utils.general import LOGGER, check_version, colorstr, resample_segments, segment2box, xywhn2xyxy
|
14 |
-
from utils.metrics import bbox_ioa
|
15 |
|
16 |
IMAGENET_MEAN = 0.485, 0.456, 0.406 # RGB mean
|
17 |
IMAGENET_STD = 0.229, 0.224, 0.225 # RGB standard deviation
|
|
|
10 |
import torchvision.transforms as T
|
11 |
import torchvision.transforms.functional as TF
|
12 |
|
13 |
+
from yolov5.utils.general import LOGGER, check_version, colorstr, resample_segments, segment2box, xywhn2xyxy
|
14 |
+
from yolov5.utils.metrics import bbox_ioa
|
15 |
|
16 |
IMAGENET_MEAN = 0.485, 0.456, 0.406 # RGB mean
|
17 |
IMAGENET_STD = 0.229, 0.224, 0.225 # RGB standard deviation
|