无左无右-CSDN博客

原创 libtorch 常用api函数示例（史上最全、最详细）

其实pytorch的函数libtorch都有，只是写法上有些出入。libtorch的官方文档链接：https文章目录调试技巧：CMakeLists.txt0.torch::full_like1.创建与初始化tensor 1.1 torch::rand 1.2 torch::empty 1.3 torch::ones 1.4 torch::Tensor keep = torch::zeros({scores.size(0)}).to(torch::kLong).to(scores.device())

2020-05-18 18:54:35 7535 13

原创匈牙利算法--任务分配

https://blog.csdn.net/ljjjjjjjjjjj/article/details/123261360例如有3个任务ABC，要分配给甲乙丙三人分别去完成，每个人完成3个任务所耗费精力不同(因为每个人特长不同)，此处也叫完成任务耗费的代价，合理分配任务，可以达到总效率最高的目标。此时若想达到耗费总精力最小，可以用穷举法一个个试，一共有6种组合，分别是：1：甲A，乙B，丙C ...

2024-03-08 16:09:00 331

原创 Python zip函数及用法

zip() 函数是 Python 内置函数之一，它可以将多个序列(列表、元组、字典、集合、字符串以及 range() 区间构成的列表)“压缩”成一个 zip 对象。所谓“压缩”，其实就是将这些序列中对应位置的元素重新组合，生成一个个新的元组。import numpy as npmy_list = [11,12,13]my_tuple = (21,22,23)my_tuple1 = (0,...

2024-03-05 17:44:00 282

原创有个shape为[7, 49, 2]的numpy矩阵，和一个shape为[6]的numpy矩阵，如何得到shape为[7, 6, 49, 3]的矩阵...

chatgpt回答：要得到一个形状为 [7, 6, 49, 3] 的 NumPy 矩阵，你需要组合这两个原始的矩阵(假设为矩阵 A 和矩阵 B)。不过，由于这两个矩阵的形状分别是 [7, 49, 2] 和 [6]，它们无法直接通过常规的广播规则来获得一个 [7, 6, 49, 3] 形状的矩阵。但根据题目的需求，我们可以用以下步骤通过拼接或堆叠来创建新的矩阵：1. 将矩阵 B 的形状变为 [7...

2024-02-29 15:02:00 374

原创 repeat得到的是[b0 b1 b0 b1]现在需要[b0 b0 b1 b1]

pytorch 一个tensor 比如是[b0b1]用tensor.repeat(2)函数可以得到[b0b1b0b1]我现在想得到[b0b0b1b1]如何优雅的得到？import torchc = torch.randint(0, 9, (2, 3))d = c.repeat(3, 1)print(f"c={c}\nd={d}")d = c.unsqu...

2024-02-19 16:48:00 288

原创 pytorch gather函数

转载于：https://www.zhihu.com/question/562282138/answer/2947708508?utm_id=0官方文档链接：https://pytorch.org/docs/stable/generated/torch.gather.html#torch.gathertorch.gather()的定义非常简洁：在指定dim上，从原tensor中获取指定ind...

2024-02-04 21:20:00 378

原创 python 列表推导式

可以看到使用列表推导式可以得到2倍速的提高ex0##self.CLASSES ('car', 'truck', 'trailer', 'bus', 'construction_vehicle', 'bicycle', 'motorcycle', 'pedestrian', 'traffic_cone', 'barrier')self.cat2id = {name: i for i, n...

2024-01-17 17:51:00 364

原创杀死僵尸显存程序 ps -ef | grep main | awk '{print $2}' |xargs kill -9

ps -ef可以看到程序名字ps -ef | grep main | awk '{print $2}' |xargs kill -9

2024-01-11 16:36:00 329

原创 Cannot re-initialize CUDA in forked subprocess.

"Cannot re-initialize CUDA in forked subprocess. " + msg)RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start methodanconda...

2024-01-09 15:29:00 442

原创 python 用uuid 生成随机字符串以命名文件确保文件名不相同

import uuidimport randoma = str(uuid.uuid1()).replace("-","")b = ''.join(random.sample(a, 6))print(uuid.uuid1())

2024-01-05 21:45:00 340

原创 vscode 代码无法转定义

方法1：试着装其他版本的python或者pylance，有时候好使有时候不好使方法2：在当前目录下的.vscode文件夹下新建settings.json文件，在里面新增：{"python.jediEnabled": false}关掉vscode项目再重打开，就可以了...

2023-12-29 10:35:00 331

原创 pytorch collate_fn测试用例

collate_fn 函数用于处理数据加载器(DataLoader)中的一批数据。在PyTorch中使用 DataLoader 时，通过设置collate_fn，我们可以决定如何将多个样本数据整合到一起成为一个 batch。在某些情况下，该函数需要由用户自定义以满足特定需求。import torchfrom torch.utils.data import Dataset, DataLoade...

2023-11-01 14:11:00 336

原创 pytorch F.grid_sample

import torchfrom torch.nn import functional as Finp = torch.ones(1, 1, 4, 4)inp = torch.randint(1, 10, (1, 1, 4, 4)).float()# 目的是得到一个长宽为20的tensorout_h = 20out_w = 20 # grid的生成方式等价于用mesh_gri...

2023-10-07 19:19:00 151

原创模型转onnx遇到问题，报错 1. _thnn_fused_lstm_cell ， 2._thnn_fused_gru_cell， 3. Exporting the operator numpy_...

目录GRULSTMExporting the operator numpy_T to ONNX opset version 11 is not supportedRuntimeError: Exporting the operator _thnn_fused_lstm_cell to ONNX opset version 11 is not supported. Please open a b...

2023-09-22 13:18:00 191

原创 python 获取时间，用时间命名

from datetime import datetimetime_str = datetime.now().strftime('%Y%m%d%H%M%S')time_str = datetime.datetime.now().strftime('%Y-%m-%d-%H-%M-%S')

2023-09-18 16:53:00 75

原创 ubuntu linux 修改密码 -->> sudo passwd 用户名

1大家用Ubuntu设置密码是会遇到一个痛点，假如设置简单的密码时，会提示让设置更复杂的密码，下面通过一种方法可以设置简单密码。sudo passwd 用户名 --->输入原来密码 --->输入要更改密码注意：锁屏密码与此是同一个sudo passwd 用户名 #这是修改开机锁屏密码，用户名就是你拷贝过来用户名sudo passwd root #这是修改root密码...

2023-09-07 16:54:00 408

原创 docker 内apt-get安装软件都不好使

报各种错误apt-get install --no-install-recommends libboost-all-devReading package lists... DoneBuilding dependency tree Reading state information... DoneThe following additional packages will b...

2023-08-22 13:09:00 529

原创 docker 报错

问题说明：我是服务器上面的docker拉到本地30卡想用的，但是失败，报错如下：服务器上面显存驱动是450，本地30卡驱动是470nvidia-docker run -it --name 20230821_3 --shm-size 16g -p 10029:22 --privileged 20230821_img bashdocker: Error response from daemon...

2023-08-21 21:04:00 280

原创 a[::-1, ...][:, ::-1] np一些转置写法

import numpy as npa = np.random.randint(0, 100, (5, 3))b = a[::-1, ...][:, ::-1]print(a)print("==" * 5)print(a[::-1, ...])print("==" * 5)print(b)print("==" * 5)c = a[::-1, ::-1, ...]print...

2023-08-08 17:35:00 60

原创平常工作中常用的linux指令

目录rsyncscpfindlocategrep -rn "123"pwd 显示当前路径rm -rf XXXdu -h 显示占用磁盘大小ln 软链接which whereisrsyncrsync -avp --exclude "*.bin" --exclude "*raw.jpg" /data/your_path [email protected]:/admin/data_1...

2023-07-28 10:31:00 55

原创 ubuntu 命令调节显示器亮度

1.显示器名字xrandr -q | grep " connected"HDMI-0 connected primary 1920x1080+0+0 (normal left inverted right x axis y axis) 527mm x 297mmDP-4 connected 1920x1080+1920+0 (normal left inverted right x axi...

2023-07-27 17:43:00 1577

原创 pycharm 远程连接服务器并且debug, 支持torch.distributed.launch debug

未经允许，本文不得转载，vx：837007389。

2023-07-27 13:18:50 3030 3

原创 linux strings 用法，解决类似libstdc++.so.6: version `CXXABI_1.3.11'问题

参考https://blog.csdn.net/Aliven888/article/details/122428668./bin/horizon: /usr/lib/libstdc++.so.6: version `CXXABI_1.3.11' not found (required by /userdata/apa/lib/libadmultivioplugin.so.1)分析：x3 板...

2023-07-26 17:38:00 283

原创一行代码python

不含有888的返回list_str = ["11_888_12", "11_888_2", "11_777_12"] bb = [x for x in list_str if '888' not in x]list_str = ["11_888_12", "11_888_2", "11_777_12"]list_str_np = np.array(list_str)aa = li...

2023-07-21 16:41:00 60

原创同时输出日志到终端和保存本地

./main | tee 111.out

2023-07-12 16:15:00 99

原创 tmux使用指令

nohup python main.py &进入后台的终端tmux attach -t yhl查看所有的tmux ls新开终端tmux new -s yhl一个终端分上下2个，进入下面的终端ctrl + b, 箭头下离开当前终端，后台运行tmax detach 或者按键 ctrl + b d关闭终端tmux kill-session -t 或者ctrl ...

2023-06-30 16:36:00 199

原创 python 多进程，进程池示例

本示例这里进程池设定的pool_num为4，但是num_task是12，也就是说4个工人同时在总共4个活，还有8个活就在排队，哪个工人干好了就赶紧领取下个活继续干，如此直到活全部干完为止。# -*- coding:utf-8 -*-from multiprocessing import Poolimport osimport timeclass AA(object): ...

2023-06-20 20:43:00 96

原创 pytorch ddp 范例

pytorch ddp 范例：################## main.py文件import argparsefrom tqdm import tqdmimport torchimport torchvisionimport torch.nn as nnimport torch.nn.functional as F# 新增：import torch.distribute...

2023-06-09 16:52:00 853 1

原创 lss

def create_frustum(self): # make grid in image plane ogfH, ogfW = self.data_aug_conf['final_dim'] #ogfH:128 ogfW:352 fH, fW = ogfH // self.downsample, ogfW // self.downsample ...

2023-06-02 17:57:00 132

原创 nuscene 数据集

nuscenes数据集官网：https://nuscenes.org/from nuscenes.nuscenes import NuScenesnusc = NuScenes(version='v1.0-mini', dataroot='/media/algo/data_1/project_others/0000paper/lss/nuScenes/mini/', verbose=True...

2023-05-30 17:37:00 265

原创 conda 安装pytorch新环境

1. 创建环境conda create -n pytorch1.7.1 python=3.72. 安装pytorch安装pytorch官网上面安装指令https://pytorch.org/get-started/previous-versions/conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=11.0 -c p...

2023-05-06 13:42:00 208 1

原创 c++ 小函数

目录1. ubuntu c++ 获取文件夹以及子文件夹下所有文件目录2. str_replace1. ubuntu c++ 获取文件夹以及子文件夹下所有文件目录#include <dirent.h>#include <vector>std::vector<std::string> GetFiles(const std::string& sdir...

2023-05-05 10:55:00 72

原创世界坐标系、相机坐标系和图像坐标系的转换

本文转载于：https://blog.csdn.net/weixin_38842821/article/details/125933604?spm=1001.2014.3001.55061. 世界坐标世界坐标系是一个特殊坐标系，它建立了描述其他坐标系需要的参考框架。能够用世界坐标系描述其他坐标系的位置，而不能用更大的、外部的坐标系描述世界坐标系。从非技术意义上讲，世界坐标系建立的是我们所关心...

2023-05-01 16:22:00 1015

原创 pytorch_ssd 代码注释

box_utils.py# -*- coding: utf-8 -*-import torchdef point_form(boxes): """ Convert prior_boxes to (xmin, ymin, xmax, ymax) representation for comparison to point form ground truth data. ...

2023-04-21 10:36:00 127

原创 numpy.random.choice(a, size=None, replace=True, p=None)

import numpy as npimport randomlist_a = ["a", "b", "c", "d", "e", "f", "g"]get_ = np.random.choice(list_a, 4, replace=False)print(get_)输出：['a' 'f' 'c' 'e']numpy.random.choice(a, size=None, re...

2023-04-11 17:57:00 66

原创终于知道centernet的sigmoid为什么需要加clamp了

终于知道centernet的sigmoid为什么需要加clamp了，由于我工程训练报错debug才知道，在计算focalloss的时候，model_out = model_out.sigmoid()pos_loss = -torch.log(model_out) * torch.pow(1 - model_out, self.alpha) * pos_inds这里当网络model_out...

2023-04-11 10:08:00 141

原创不定时更新解决的一些问题

目录修改docker时间修改docker时间https://blog.csdn.net/tiny_du/article/details/124024562复制宿主机上的zoneinfo文件夹到容器下的/usr/share/目录下。docker cp /usr/share/zoneinfo 容器名称:/usr/share/进入到容器中docker exec -it 容器名称 /bin/...

2023-04-03 13:38:00 48

原创清理僵尸显存进程

nvidia-smi命名看不到进程，但是显存占用着这个时候执行fuser -v /dev/nvidia*然后一个个的杀掉进程

2023-04-03 09:30:00 264

原创 ubuntu两台电脑之间硬盘远程挂载

目录服务器电脑配置：1. 安装nfs服务：sudo apt-get install nfs-kernel-server rpcbind2. 查看nfs配置文件：sudo vi /etc/exports3. 修改nfs配置文件：sudo vi /etc/exports4. 重启nfs服务器：sudo /etc/init.d/nfs-kernel-server restart5. 检查nfs服务器挂...

2023-03-30 13:32:00 1268 1

原创解决vscode由于sys.path.insert(0, './utils/')导致无法转定义问题

由于代码用到了当前utils文件夹下的文件里面函数。from xml_utils import load_xml_and_parse_new这里xml_utils是utils文件夹下的xml_utils.py。然后无法转定义很是麻烦，搞了半天没找到方法，然后当我把鼠标放在波浪号上面的时候会显示：点快速修复：然后加入就解决了。会发现在当前文件夹生成了./.vscode/settings...

2023-03-28 13:44:00 182