实用代码
ssf-yasuo
把博客当笔记本用了,私信和评论可能很久才会回复
展开
-
计算psnr ssim niqe fid mae lpips等指标的代码
【代码】计算psnr ssim niqe fid mae lpips等指标的代码。原创 2024-04-10 21:02:17 · 337 阅读 · 0 评论 -
KMP算法代码
【代码】KMP算法代码。原创 2024-03-21 22:10:10 · 103 阅读 · 0 评论 -
A*算法python实现
【代码】A*算法python实现。原创 2024-03-19 12:47:00 · 379 阅读 · 0 评论 -
RUAS代码debug
最近训练Retinex-inspired Unrolling with Cooperative Prior Architecture Search for Low-light Image Enhancement的代码(https://github.com/KarelZhang/RUAS)的时候报错说梯度回传有问题,有变量被修改了,仔细一看源码,问题出在这里:model.py文件中,226行是报错位置,也就是说denoise_loss backward的时候,与denoise_loss有梯度回传关系的张原创 2023-04-14 16:39:59 · 440 阅读 · 1 评论 -
exdark数据集转yolo格式(仅供参考)
exdark数据集转yolo格式原创 2022-07-26 22:12:33 · 2542 阅读 · 23 评论 -
pytorch版本兼容问题
今天用pytorch1.8跑官方版本Neighbor2Neighbor的时候报了这个错:TypeError: conv_transpose2d(): argument 'output_padding' (position 6) must be tuple of ints, not tuple发现问题出在nn.ConvTranspose2d上,代码是这样写的:self.deconv = nn.ConvTranspose2d(in_nc, out_nc, 2, 2, 0, False)很奇.原创 2022-05-28 11:21:10 · 2674 阅读 · 12 评论 -
python 读图片的shape问题
img = Image.open读取图片后,img.size返回的是w,h,而进一步npimg = np.array(img)后,npimg.shape返回的是h,w,c,如果读取的图片带透明度通道,c还可能是4,open的时候如果用了convert(‘RGB’),才会变成3。而当使用img = Image.fromarray(npimg.astype(np.uint8))后,size又会回到w, h...原创 2022-05-22 12:14:23 · 1946 阅读 · 0 评论 -
widerface数据集转yolo格式
widerface 数据集 转 yolo 格式原创 2022-03-14 15:52:14 · 2175 阅读 · 5 评论 -
darkface 转 yolo 格式
darkface 转yolo格式代码原创 2022-03-12 23:14:42 · 4717 阅读 · 1 评论 -
python 添加 import 的路径
import其它路径的模块时可以用到的方法原创 2022-02-19 13:05:28 · 2479 阅读 · 0 评论 -
pytorch dataloader collate_fn 在复杂label情况下的使用技巧
pytorch dataset 在目标检测中使用 collate_fn来打包具有不同个数bbox的标注原创 2022-02-19 11:36:31 · 2604 阅读 · 0 评论 -
exdark数据集转coco格式
exdark数据集转coco格式原创 2022-01-28 15:39:44 · 1256 阅读 · 14 评论 -
coco转yolo txt格式
"""author: Wu2021/1/24source: https://github.com/Weifeng-Chen/DL_tools/blob/main/coco2yolo.pyCOCO 格式的数据集转化为 YOLO 格式的数据集,源代码采取遍历方式,太慢,这里改进了一下时间复杂度,从O(nm)改为O(n+m),但是牺牲了一些内存占用--json_path 输入的json文件路径--save_path 保存的文件夹名字,默认为当前目录下的labels。"""import os原创 2022-01-24 20:15:09 · 318 阅读 · 0 评论 -
python多线程图片处理
有时候我们需要对图片进行如加噪、放缩等变换,当数据集比较大时,遍历整个数据集所用时间太长,为此可以写一点多线程处理的代码,下面以压缩为例子:# author: Wu# 2022/1/24# This script is used to jpeg compressimport osimport cv2import numpy as npfrom threading import Threadinput_folder = '/data0/wjh/coco/val2017'output_.原创 2022-01-24 18:37:03 · 1864 阅读 · 0 评论 -
利用cocoapi计算json 预测的map
用yolov4跑了下 coco 的 val2017,想看看map,但是coco2017的json标注又不能直接用,所以先用yolov4的./darknet detector valid data/coco.data cfg/yolov4.cfg backup/yolov4.weights生成result.json然后再用以下脚本计算map# author: Wu# modified from: https://blog.csdn.net/qq_40527427/article/details/1原创 2022-01-22 22:28:23 · 1284 阅读 · 2 评论 -
pytorch单GPU模型加载多GPU保存的参数
from collections import OrderedDictstate_dict = torch.load(para_path)new_state_dict = OrderedDict()for k, v in state_dict.items(): name = k[7:] # remove `module.` new_state_dict[name] = vnet.load_state_dict(new_state_dict)原创 2021-11-24 15:08:54 · 1081 阅读 · 0 评论 -
将NWPU VHR-10数据集转换为coco格式
"""Created on 5/11读取txt文件,划分训练集和测试集并且生成coco格式的json文件@author: Wu"""import jsonimport osimport numpy as npimport pandas as pdimport reimport cv2import shutilfrom PIL import Imagefrom sklearn.model_selection import train_test_split# generate cl原创 2020-05-12 14:43:23 · 1557 阅读 · 2 评论 -
将NWPU VHR-10 数据集转换为yolo的数据集格式
import osimport pandas as pdimport cv2ann_path = 'E:/dataset/NWPU VHR-10/NWPU VHR-10 dataset/ground truth'output_path = 'E:/dataset/NWPU VHR-10/NWPU VHR-10 dataset/yolo-format'im_path = 'E:/dataset/NWPU VHR-10/NWPU VHR-10 dataset/positive image set'a原创 2020-05-12 14:41:56 · 1865 阅读 · 4 评论 -
python 打印表格(如深度学习想打印混淆矩阵)
来源:prettytable confusion_matrix = [[0 for j in range(4)] for i in range(4)] for i, j in zip(new_predicted, new_targets): confusion_matrix[i][j] += 1 tb = pt.PrettyTable( ["predicted\\...原创 2020-04-01 18:58:02 · 719 阅读 · 0 评论 -
画froc曲线代码,供参考
import pandas as pdimport numpy as np import jsonimport matplotlib.pyplot as pltfrom decimal import Decimalfrom matplotlib.ticker import FixedFormatterimport osresult_csv_file_path = 'xxx.csv'...原创 2020-03-10 19:47:29 · 1107 阅读 · 3 评论 -
将deeplesion数据集转化为coco格式的检测数据集,可参考
"""Created on 2/261. read csv,save as cocostyle dataset:--dataset/ --trainset/ --image/ --***.jpg ***.jpg ... --annotation/ --annotation.json --validset/ -...原创 2020-03-10 19:42:03 · 984 阅读 · 4 评论 -
根据deeplesion给的官方标注文件的 Measurement_coordinates 生成伪mask的代码
import numpy as npimport pandas as pdimport cv2import osann_file_path = 'xxx/DL_info.csv'output_path = 'xxx/fake_masks'def findIntersection(list_in): vector0 = [] vector1 = [] vect...原创 2020-03-10 17:16:47 · 258 阅读 · 0 评论 -
根据分割图生成bbox的csv标注:
根据分割图生成bbox的csv标注:import numpy as npimport skimagefrom skimage.measure import label,regionpropsfrom skimage.filters import robertsimport osimport cv2import pandas as pddef get_patch_csv(filen...原创 2020-03-10 16:56:22 · 541 阅读 · 9 评论 -
linux操作笔记
linux 查看当前路径下文件和文件夹 lslinux统计当前路径下文件数量(非递归)ls -l |wc -llinux后台运行 nohup somecommand > log 2>&1 &复制:cp删除:rm移动/重命名:mv显示当前路径 : pwd*代表匹配到的所有文件,比如,rm * 代表删除当前路径下所有文件,rm *.jpg代表删除当前路径下...原创 2020-03-10 15:24:43 · 137 阅读 · 0 评论 -
医学图像CT预处理,分割肺部(不只是肺实质,整个肺)
肺部多病种研究需要同时检测肺实质中的结节、索条和心脏、血管中的动脉硬化,而网上只存在肺实质的提取,索性自己写一个,先阈值化,后对背景进行flood_fill算法再取反,同时用到了一些腐蚀膨胀操作。不得不说,skimage真好用,其实skimage也有floodfill,但是不知道为什么出了bug,所以用了cv2的floodfillimport numpy as npimport skimage...原创 2019-11-12 23:16:53 · 7149 阅读 · 18 评论