大白成长记-CSDN博客

转载 RuntimeError: cuda runtime error (59) : device-side assert triggered

RuntimeError: cuda runtime error (59) : device-side assert triggered

2020-10-30 10:13:14 173

转载 pytorch中torch.nn.dropout和torch.nn.F.dropout区别

链接地址

2020-08-09 09:24:32 1698 1

转载训练model过程中，loss=nan问题时该怎么解决？

当我在UCF-101数据集训练alexnet时，epoch设为100，跑到三十多个epoch时，出现了loss=nan问题，当时是一脸懵逼，在查阅资料后，我通过减小学习率解决了问题，现总结一下出现这个问题的可能原因及解决方法：1. 减小整体学习率。学习率比较大的时候，参数可能over shoot了，结果就是找不到极小值点；减小学习率可以让参数朝着极值点前进；2. 改变网络宽度。有可能是网络后面的层参数更新异常，增加后面层的宽度试试；3. 改变层的学习率。每个层都可以设置学习率，可以尝试减小后...

2020-08-08 18:57:44 946

转载 ubuntu 安装多版本cuda(原9.0，新安装10.0)，并随时切换使用

ubuntu安装多版本cuda

2020-07-21 19:14:38 738

转载协同过滤介绍

协同过滤介绍

2020-06-11 09:07:57 219

转载 CenterNets代码解析

代码解析1代码解析2

2020-06-02 17:07:34 1197

转载 Maximum Classifier Discrepancy for Unsupervised Domain Adaptation

Transferability vs. Discriminability: Batch Spectral Penalization for Adversarial Domain AdaptationMaximum Classifier Discrepancy for Unsupervised Domain Adaptation

2020-05-28 21:37:29 326

转载 CenterNet：Objects as Points详解

CenterNet： Objects as Points分析

2020-05-26 16:09:53 495

转载目标检测评价标准mAP

1，自信度confidence只是用来排序。然后AP是由同一recall下的最高的准确度来进行面积计算来得到AP的分数。对于Recall >= {0, 0.14, 0.29, 0.43, 0.57, 0.71, 1}，我们选取此时Percision的最大值：1, 1, 1, 0.5, 0.5, 0.5, 0。计算recall/precision下的面积：AP = (0.14-0)x1 + (0.29-0.14)x1 + (0.43-0.29)x0.5 + (0.57-0.43)x0.5 + (0.7

2020-05-25 21:30:39 1711

转载 Ubuntu下U盘只读文件系统，图标上锁，提示无法修改

Linux下解决U盘只可读操作

2020-05-25 08:30:51 390

转载 conda命令的相应使用

conda命令的相应使用coda命令的使用art.js/

2020-05-22 15:07:12 243

转载 python画attention map 与原热力图叠加

python 画attention map 与原图叠加

2019-12-17 20:47:04 6236

原创 python中base feature 保存.h5

import h5pywith h5py.File('attention_map.h5','w') as f: m = torch.squeeze(base_feat.detach()) #pdb.set_trace() m=m.sum(0)/1024.0 m=m.unsqueeze(2).expand(38,53,3) #pdb.set_tr...

2019-12-17 20:44:37 244

原创 python中缩小图片并保持

# -*- coding: utf-8 -*-from torchvision import transformsfrom PIL import Imageimport warningswarnings.filterwarnings("ignore")img = Image.open('/home/images/969743078.jpg')print(type(img)...

2019-12-17 20:42:26 1057