- 博客(397)
- 资源 (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 7497 13
原创 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 283
原创 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 367
原创 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 357
原创 杀死僵尸显存程序 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 328
原创 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 424
原创 python 用uuid 生成随机字符串以命名文件确保文件名不相同
import uuidimport randoma = str(uuid.uuid1()).replace("-","")b = ''.join(random.sample(a, 6))print(uuid.uuid1())
2024-01-05 21:45:00 317
原创 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 327
原创 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 149
原创 模型转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 175
原创 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 70
原创 ubuntu linux 修改密码 -->> sudo passwd 用户名
1大家用Ubuntu设置密码是会遇到一个痛点,假如设置简单的密码时,会提示让设置更复杂的密码,下面通过一种方法可以设置简单密码。sudo passwd 用户名 --->输入原来密码 --->输入要更改密码 注意:锁屏密码与此是同一个sudo passwd 用户名 #这是修改开机锁屏密码,用户名就是你拷贝过来用户名sudo passwd root #这是修改root密码...
2023-09-07 16:54:00 375
原创 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 511
原创 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 271
原创 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 59
原创 平常工作中常用的linux指令
目录rsyncscpfindlocategrep -rn "123"pwd 显示当前路径rm -rf XXXdu -h 显示占用磁盘大小ln 软链接which whereisrsyncrsync -avp --exclude "*.bin" --exclude "*raw.jpg" /data/your_path algo@192.168.192.38:/admin/data_1...
2023-07-28 10:31:00 54
原创 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 1537
原创 pycharm 远程连接服务器并且debug, 支持torch.distributed.launch debug
未经允许,本文不得转载,vx:837007389。
2023-07-27 13:18:50 2735 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 272
原创 一行代码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 57
原创 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 196
原创 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 94
原创 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 843 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 131
原创 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 257
原创 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 196 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 69
原创 世界坐标系、相机坐标系和图像坐标系的转换
本文转载于:https://blog.csdn.net/weixin_38842821/article/details/125933604?spm=1001.2014.3001.55061. 世界坐标世界坐标系是一个特殊坐标系,它建立了描述其他坐标系需要的参考框架。能够用世界坐标系描述其他坐标系的位置,而不能用更大的、外部的坐标系描述世界坐标系。从非技术意义上讲,世界坐标系建立的是我们所关心...
2023-05-01 16:22:00 988
原创 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 121
原创 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 64
原创 终于知道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 135
原创 不定时更新解决的一些问题
目录修改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 1230 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 179
原创 pytorch gather b2 = a.gather(1, b.view(-1, 1))
import torcha = torch.randint(0, 100, (6,3))b = torch.Tensor([0, 1, 1, 2, 0, 2]).long()b = b.unsqueeze(1)b0 = b.view(-1, 1)b2 = a.gather(1, b.view(-1, 1))print(a)print(a.shape)print(b)pri...
2023-03-25 19:23:00 67
原创 caffe 显示AnnotatedDatum代码
static bool show_rt(cv::Mat &show, cv::Rect &rt, int label) { static std::string label_map[] = { "bg", "Pedestrian...
2023-03-24 14:33:00 36
原创 caffe中多个cpp共享一个变量 c++类中的静态变量
caffe中需要整个共享变量,就是从bias过来的tensor转图片,然后后面目标检测第一阶段查看定位效果,把目标框画在图上就需要一开始的图片。实验1:这个可以在prototxt增加一个top通过网络给到后面的层,但是这样太麻烦。实验2:然后想想能不能通过类似与共享变量能完成,首先实验的是在基类Layer添加一个变量,然后bias_layer压入图片,因为所有的层都是从Layer派生出来...
2023-03-24 09:50:00 174
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关注的人