- 博客(11)
- 收藏
- 关注
原创 复现OC-sort并在自己的数据集上跑通
复制一份yolox_dancetrack.py改名字为yolox_track.py,根据自己数据的属性更改其中参数。只在dancetrack上复现(Windows 系统,显卡4090, torch1.13.1+cu117)复现自己的数据(Windows 系统,显卡4090, torch1.13.1+cu117)更改路径DATA_PATH为存放文件夹的绝对路径,右击run下面代码,数据存放如下图, 每个文件中的图像序列为一个连续的视频采集。1.准备自己的数据,准备数据的标签为yolov5格式。
2024-06-18 17:14:57
444
2
原创 全连接层计算原理
该全连接层的参数量为3*2+2=8分解为权重数量W加上偏置数量B。计算可以为前一层神经元数量*当前层的数量(输出层的数量)+ 偏置数量(与输出层的数量相等)计算过程为y1 = (x1*w11 + b11) + (x2*w21 + b21) + (x1*w31 + b31)假设:输入的神经元长度为3。x1, x2, x3,经过全连接操作后输出的神经元长度为2。计算过程将特征F展平为 c * w * h长度的序列然后通过全连接计算为改变序列长度为3。
2024-02-20 15:04:48
630
原创 Process finished with exit code 139 (interrupted by signal 11: SIGSEGV)
解决方法:安装包冲突,没找到具体哪个包,重新创建环境解决问题。
2022-10-04 21:56:35
272
1
原创 ** ERROR (nifti_image_write_hdr_img2): cannot open output file ‘fixed_lits/label\hepaticvessel_001.n
解决方法:子目录不存在,写入的时候未自动创建这个子目录。
2022-10-01 22:21:07
302
原创 NCM模块
import torchimport torch.nn as nnclass NCM(nn.Module): def __init__(self, in_ch, out_ch): super(NCM, self).__init__() self.con1x1 = nn.Sequential(nn.Conv2d(in_ch, out_ch, kernel_size=(1, 1), stride=(1, 1), padding=(0, 0)), .
2022-05-23 18:21:03
22
原创 pytorch实现CBAM模块
# ------------------------## CBAM模块的Pytorch实现(一般加入中间层效果较好)# ------------------------#import torchimport torch.nn as nn# 通道注意力模块class ChannelAttentionModule(nn.Module): def __init__(self, channel=256, reduction=16): super(ChannelAttenti.
2022-05-07 10:19:43
840
原创 pytorch-lightning网络生成的张量,转换为JPG文件,并保存
def test_step(self, batch, batch_idx): ct, mask, name = batch outputs = self.forward(ct) self.measure(batch, outputs) print(ct.shape)########################################################################################### for outp.
2022-05-06 18:29:03
549
原创 python加载npz或npy文件 转换为深度图和灰度图
加载npz文件生成灰度图和深度图:import osfrom glob import globimport cv2import imageioimport matplotlib.pyplot as pltimport numpy as npinput_path = "kits_tumor/pred"paths = glob(os.path.join(input_path,"*_*.npz"))out_path = "save_path"for path in os.listdi
2022-05-05 16:55:30
2799
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人