Spaces:
Build error
Build error
onipot
commited on
Commit
·
dd4f531
1
Parent(s):
a254402
update augs
Browse files- yolov5/detect.py +2 -1
yolov5/detect.py
CHANGED
|
@@ -38,6 +38,8 @@ test_random_transforms = torchvision.transforms.Compose([
|
|
| 38 |
torchvision.transforms.RandomGrayscale(p=0.4),
|
| 39 |
torchvision.transforms.RandomPerspective(0.4, p=0.4),
|
| 40 |
torchvision.transforms.RandomAdjustSharpness(2),
|
|
|
|
|
|
|
| 41 |
torchvision.transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)),
|
| 42 |
torchvision.transforms.Resize((224, 224)),
|
| 43 |
])
|
|
@@ -96,7 +98,6 @@ def predict(
|
|
| 96 |
# Process predictions
|
| 97 |
preds = []
|
| 98 |
|
| 99 |
-
|
| 100 |
for i, det in enumerate(pred): # per image
|
| 101 |
|
| 102 |
# im0 = im0.copy()
|
|
|
|
| 38 |
torchvision.transforms.RandomGrayscale(p=0.4),
|
| 39 |
torchvision.transforms.RandomPerspective(0.4, p=0.4),
|
| 40 |
torchvision.transforms.RandomAdjustSharpness(2),
|
| 41 |
+
torchvision.transforms.RandomAffine(degrees=0, translate=None, scale=(0.9, 1.0)),
|
| 42 |
+
torchvision.transforms.RandomHorizontalFlip(),
|
| 43 |
torchvision.transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)),
|
| 44 |
torchvision.transforms.Resize((224, 224)),
|
| 45 |
])
|
|
|
|
| 98 |
# Process predictions
|
| 99 |
preds = []
|
| 100 |
|
|
|
|
| 101 |
for i, det in enumerate(pred): # per image
|
| 102 |
|
| 103 |
# im0 = im0.copy()
|