Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

New scale_segments() function (#9570)

Browse Source 
* Rename scale_coords to scale_boxes

* add scale_segments
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-09-24 16:02:41 +02:00
1 parent d669a74623
commit c8e52304cf
8 files changed
+54 -28
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

utils/loggers/comet/__init__.py
+4 -4
View File
@@ -28,7 +28,7 @@ import torchvision.transforms as T
import yaml
from utils.dataloaders import img2label_paths
from utils.general import check_dataset, scale_coords, xywh2xyxy
from utils.general import check_dataset, scale_boxes, xywh2xyxy
from utils.metrics import box_iou
COMET_PREFIX = "comet://"
@@ -293,14 +293,14 @@ class CometLogger:
pred[:, 5] = 0
predn = pred.clone()
scale_coords(image.shape[1:], predn[:, :4], shape[0], shape[1])
scale_boxes(image.shape[1:], predn[:, :4], shape[0], shape[1])
labelsn = None
if nl:
tbox = xywh2xyxy(labels[:, 1:5]) # target boxes
scale_coords(image.shape[1:], tbox, shape[0], shape[1]) # native-space labels
scale_boxes(image.shape[1:], tbox, shape[0], shape[1]) # native-space labels
labelsn = torch.cat((labels[:, 0:1], tbox), 1) # native-space labels
scale_coords(image.shape[1:], predn[:, :4], shape[0], shape[1]) # native-space pred
scale_boxes(image.shape[1:], predn[:, :4], shape[0], shape[1]) # native-space pred
return predn, labelsn