From b00ab5c658c1b06c8d9aacab34a750ea134c0ba0 Mon Sep 17 00:00:00 2001 From: Li Bin Date: Fri, 26 May 2023 10:43:38 +0800 Subject: [PATCH] fixed linter --- mmdet3d/datasets/convert_utils.py | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/mmdet3d/datasets/convert_utils.py b/mmdet3d/datasets/convert_utils.py index 1d2dda7772..2f765fc644 100644 --- a/mmdet3d/datasets/convert_utils.py +++ b/mmdet3d/datasets/convert_utils.py @@ -1,6 +1,7 @@ # Copyright (c) OpenMMLab. All rights reserved. import copy from typing import List, Optional, Tuple, Union +import warnings import numpy as np from nuscenes import NuScenes @@ -360,18 +361,18 @@ def post_process_coords( if polygon_from_2d_box.intersects(img_canvas): img_intersection = polygon_from_2d_box.intersection(img_canvas) if isinstance(img_intersection, Polygon): - intersection_coords = np.array( - [coord for coord in img_intersection.exterior.coords]) - min_x = min(intersection_coords[:, 0]) - min_y = min(intersection_coords[:, 1]) - max_x = max(intersection_coords[:, 0]) - max_y = max(intersection_coords[:, 1]) - return min_x, min_y, max_x, max_y + intersection_coords = np.array( + [coord for coord in img_intersection.exterior.coords]) + min_x = min(intersection_coords[:, 0]) + min_y = min(intersection_coords[:, 1]) + max_x = max(intersection_coords[:, 0]) + max_y = max(intersection_coords[:, 1]) + return min_x, min_y, max_x, max_y else: - warnings.warn('img_intersection is not an object of Polygon.') - return None + warnings.warn('img_intersection is not an object of Polygon.') + return None else: - return None + return None def generate_record(ann_rec: dict, x1: float, y1: float, x2: float, y2: float,