主音贝斯手-优快云博客

原创 3D目标检测算法调研&FCOS/FCOS3D/FCOS3D++算法介绍

简单介绍3D目标检测发展过程中的部分重要算法，以及FCOS到FCOS3D及FCOS3D++的重要演变过程。

2022-11-21 10:53:33 7040 1

原创 pytorch→ONNX→C++部署注意事项

torch.onnx.export（）函数中export_params=True，不能是False。

2021-02-22 15:46:40 858

原创 C++：Libtorch在win10系统visual studio 2019中的配置方法

1、在https://pytorch.org中下载对应版本的libtorch，这里以1.6的cpu+release版本为例进行说明（图中为1.7，笔者用的1.6）。2、解压后于环境变量的path中添加xxx\libtorch\lib。3、打开vs2019建立新项目，选择release和x64模式，属性管理器中在release|x64中添加新项目属性表起名libtorch（随便起一个）。4、包含目录添加：xxx\libtorch\includexxx\libtorch\include\torch\

2020-11-23 15:59:35 1443

原创 C：读取文本文件每一行并删除末尾换行符

#define _CRT_SECURE_NO_WARNINGS#include <stdio.h> int main() { char *filename = "E://darknet-master-cpu/build/ceshi.txt"; //文件名 FILE *fp; char name[256]; //每行最大读取的字符数 fp = fopen(filename, "r");

2020-10-20 16:02:52 3911

原创 Pytorch：语义分割评价指标

在这里class SegMetric(object): def __init__(self, n_classes): self.n_classes = n_classes self.confusion_matrix = np.zeros((n_classes, n_classes)) def _fast_hist(self, label_true, label_pred, n_class): mask = (label_true >=

2020-08-10 21:42:38 1851

转载 Python：读取文件名写入txt

import osfile_path = "Annotations"# os.listdir(file)会历遍文件夹内的文件并返回一个列表path_list = os.listdir(file_path)# print(path_list)# 定义一个空列表,我不需要path_list中的后缀名path_name=[]# 利用循环历遍path_list列表并且利用split去掉后缀名for i in path_list: path_name.append(i.split(".")[

2020-06-30 10:11:04 1142

原创 Python：裁剪文件夹中的所有照片

import cv2import osrootdir = r'./before/' # 指明被遍历的文件夹for parent, dirnames, filenames in os.walk(rootdir): for filename in filenames: currentPath = os.path.join(parent, filename) img = cv2.imread(currentPath) cropped = img[1920:3360, 0:1440] #

2020-06-30 08:38:31 493

原创 Matlab：png2mat

clc;clear;namelist = dir('F:\mat\*.png');len = length(namelist); for i = 1:len png_file = fullfile( 'F:\mat\', namelist(i).name ); I = imread( png_file); % path = 'F:\mat\' name = '001' ext = '.png' [ path , name , ext ] = filep

2020-06-29 10:39:02 251

原创 Python：将彩色标注图转变为灰度值标签图

import osfrom PIL import Imageimport cv2 as cvimport numpy as nplabel_dir = './label/'label_process_dir = './label_process/'if not os.path.exists(label_process_dir): os.makedirs(label_process_dir)for file in os.listdir(label_dir): label =

2020-06-29 10:36:29 1344

原创 Python：检验PS标注是否准确

import numpy as npimport osimport cv2 as cvlabel_dir = './png/'label_process_dir = './label_process/'for file in os.listdir(label_dir): label = label_dir + file if os.path.isfile(label): print(file) # 获得一张图片的所有种类的灰度值 l

2020-06-29 10:34:09 228

原创 Python：数据增强，水平翻转/亮度饱和度调整

from PIL import Imageimport numpy as npimport osimport os.pathimport cv2MAX_VALUE = 100'''### 图片左右对称 ===rootdir = r'F:\data\png_3' # 指明被遍历的文件夹for parent, dirnames, filenames in os.walk(rootdir): for filename in filenames: print('parent i

2020-06-29 10:32:04 1513

原创 Pytorch：One-hot编码

def mask2onehot(mask, num_classes): # mask为灰度图，(H,W)→(K,H,W) _mask = [mask == i for i in range(num_classes)] return np.array(_mask).astype(np.uint8)

2020-06-29 10:15:17 774

原创 Pytorch：使用visdom读取.mat文件

from visdom import Visdomimport scipy.io as ioimport numpy as np#python -m visdom.server #于cmd输入vis = Visdom()pred = io.loadmat(‘E:/abc/123.mat’)pred = pred[‘pred’].astype(np.uint8) #'pred’为自定义的.mat内部变量pred = 100 * pred #100为了拉大灰度值差异，可自定义vis.im

2020-06-29 09:56:33 3724

原创 Pytorch：MNIST十分类

import torchimport torch.nn as nnimport torchvision.datasets as normal_datasetsimport torchvision.transforms as transformsfrom torch.autograd import Variablenum_epochs = 5learning_rate = 0.001def get_variable(x): x = Variable(x) return x.cuda()

2020-06-09 23:36:59 625

m0_37908025的博客