- 博客(400)
- 资源 (19)
- 收藏
- 关注
原创 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 7543 13
原创 匈牙利算法--任务分配
https://blog.csdn.net/ljjjjjjjjjjj/article/details/123261360例如有3个任务ABC,要分配给甲乙丙三人分别去完成,每个人完成3个任务所耗费精力不同(因为每个人特长不同),此处也叫完成任务耗费的代价,合理分配任务,可以达到总效率最高的目标。此时若想达到耗费总精力最小,可以用穷举法一个个试,一共有6种组合,分别是:1:甲A,乙B,丙C ...
2024-03-08 16:09:00 332
原创 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 290
原创 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 381
原创 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 365
原创 杀死僵尸显存程序 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 330
原创 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 450
原创 python 用uuid 生成随机字符串以命名文件确保文件名不相同
import uuidimport randoma = str(uuid.uuid1()).replace("-","")b = ''.join(random.sample(a, 6))print(uuid.uuid1())
2024-01-05 21:45:00 342
原创 vscode 代码无法转定义
方法1:试着装其他版本的python或者pylance,有时候好使有时候不好使方法2:在当前目录下的.vscode文件夹下新建settings.json文件,在里面新增:{"python.jediEnabled": false}关掉vscode项目再重打开,就可以了...
2023-12-29 10:35:00 333
原创 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 339
原创 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 152
原创 模型转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 197
原创 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 79
原创 ubuntu linux 修改密码 -->> sudo passwd 用户名
1大家用Ubuntu设置密码是会遇到一个痛点,假如设置简单的密码时,会提示让设置更复杂的密码,下面通过一种方法可以设置简单密码。sudo passwd 用户名 --->输入原来密码 --->输入要更改密码 注意:锁屏密码与此是同一个sudo passwd 用户名 #这是修改开机锁屏密码,用户名就是你拷贝过来用户名sudo passwd root #这是修改root密码...
2023-09-07 16:54:00 416
原创 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 537
原创 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 283
原创 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 61
原创 平常工作中常用的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 56
原创 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 1583
原创 pycharm 远程连接服务器并且debug, 支持torch.distributed.launch debug
未经允许,本文不得转载,vx:837007389。
2023-07-27 13:18:50 3130 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 286
原创 一行代码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 62
原创 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 857 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 270
原创 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 209 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 73
原创 世界坐标系、相机坐标系和图像坐标系的转换
本文转载于:https://blog.csdn.net/weixin_38842821/article/details/125933604?spm=1001.2014.3001.55061. 世界坐标世界坐标系是一个特殊坐标系,它建立了描述其他坐标系需要的参考框架。能够用世界坐标系描述其他坐标系的位置,而不能用更大的、外部的坐标系描述世界坐标系。从非技术意义上讲,世界坐标系建立的是我们所关心...
2023-05-01 16:22:00 1033
原创 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 144
原创 不定时更新解决的一些问题
目录修改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
原创 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 1282 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
attention-is-all-you-need-pytorch-zhushi-代码注释
2023-03-22
snake_cuda10_and_libtorch.tar
2020-10-20
snake-master-cuda8-zhuan_pt.tar
2020-10-20
libtorch1.3-cuda10-refinedet.tar.gz
2020-05-26
python opencv中文高清翻译教程
2018-09-20
Neural Network and Deep Learning高清中英文双版pdf
2018-01-16
Linux进程管理
2015-11-14
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人