- 博客(10)
- 资源 (1)
- 收藏
- 关注
原创 nuScenes注意事项
全局坐标系:方向为“东北上”,原点在“经度0°、纬度0°” 车身坐标系:方向为“前左上”,原点在“后轴中心” 相机坐标系:方向为“右下前”,原点在“镜头光心” 激光坐标系:方向为“右前上”,原点在“激光器” 毫米波坐标系:方向为“前左上”,原点在“天线中心”Radar点云以毫米波坐标系存储
2023-07-07 02:30:22
1092
1
原创 Numba注意事项
Numba对for的加速好过对NumPy的加速,建议把NumPy拆分为for循环 Numba对函数加速更方便,不要把函数作为类方法 示例 from numba import jit@jit(nopython=True)def func(x, y): return x + y
2022-03-28 23:28:15
968
原创 计算交并比
边界框的面积交并比import numpy as npdef compute_iou(boxes1, boxes2): boxes1, boxes2 = np.array(boxes1), np.array(boxes2) lt = np.maximum(boxes1[:, None, :2], boxes2[:, :2]) rb = np.minimum(boxes1[:, None, 2:], boxes2[:, 2:]) wh = np.maximum(rb
2021-12-16 21:02:13
326
原创 计算欧式距离
利用Numpy计算:import numpy as npdef euclidean_distance(row_points): distances = np.linalg.norm(row_points, ord=2, axis=1) return distancesif __name__ == '__main__': row_points = np.array([[3, 4]]).repeat(10, axis=0) distances = euclide
2021-12-04 11:08:23
818
原创 计算渐变色的RGB值
渐变色可用于画热度图、分布图等。渐变色示例:import cv2import numpy as npdef gradient_ramp(num=256): gradients = np.zeros((256, 3), dtype=np.uint8) series = np.arange(0, 256) gradients[:, 0] = series gradients[:, 2] = series[::-1] indices = np.linsp
2021-12-03 21:54:19
1768
原创 [PyTorch] Deformable Convolution示例
import torchimport torch.nn as nnimport torchvision.ops as opsclass Net(nn.Module): def __init__(self, in_c, out_c, k=3): super().__init__() p = (k - 1) // 2 self.conv_offset = nn.Conv2d(in_c, 2 * k * k, k, padding=p) .
2021-11-24 16:47:35
2941
原创 NumPy每行/每列只保留一个最大值/最小值,其他元素置为0
import numpy as npinput = np.random.randint(100, size=(3, 4))indices = np.argmax(input, axis=1)indices = np.expand_dims(indices, axis=1)output = np.zeros_like(input)np.put_along_axis(output, indices, 1, axis=1)
2021-06-15 11:49:00
2130
2
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人