- 博客(5)
- 资源 (2)
- 收藏
- 关注
原创 【YOLOv3】训练新数据集记录
参考代码:https://github.com/eriklindernoren/PyTorch-YOLOv31.训练时loss出现nan原因:默认学习率learning_rate=0.001过大导致梯度爆炸,出现inf与nan值解决办法:在yolov3.cfg中降低学习率:参考博客:网络训练时出现loss为nan的情况(已解决)神经网络训练时,出现NaN loss...
2020-11-05 15:23:47 762
原创 安装pytorch时InvalidArchiveError
报错:InvalidArchiveError("Error with archive /anaconda3/pkgs/pytorch-1.2.0-py3.7_cuda10.0.130_cudnn7.6.2_0.tar.bz2. You probably need to delete and re-download or re-create this file. Message from libarchive was:\n\nFailed to open '/anaconda3/pkgs/pytorch
2020-11-05 13:02:54 6864 1
原创 torch.t()、torch.min()、*、torch.stack、.long()
yolov3中的bbox_wh_iou代码如下:def bbox_wh_iou(wh1, wh2): wh2 = wh2.t() # w1是当前anchor的w,h1是当前anchor的h w1, h1 = wh1[0], wh1[1] # w2是所有目标的w,h2是所有目标的h w2, h2 = wh2[0], wh2[1] inter_area = torch.min(w1, w2) * torch.min(h1, h2) union_area
2020-11-04 15:36:13 1268
原创 Pytorch中的tensor[...,0]
参考博客:PyTorch中[…, 0]的使用pytorch中[…, 0]的用法>>> a= torch.rand(1,2,2,2,2)>>> atensor([[[[[0.2528, 0.9359], [0.1876, 0.4189]], [[0.5546, 0.1931], [0.1389, 0.7482]]], [[[0.6169, 0.2977],
2020-11-03 22:16:46 1634
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人