pytorch\keras
pytorch\keras
辣大辣条
这个作者很懒,什么都没留下…
展开
-
dice_loss+BCE_loss 源码
dice loss+bce loss原创 2022-11-22 14:44:28 · 905 阅读 · 1 评论 -
WeightedCrossEntropyLoss 源码
WeightedCrossEntropyLoss原创 2022-11-22 14:39:03 · 226 阅读 · 0 评论 -
OHEM loss 源代码
ohem loss原创 2022-11-22 14:36:08 · 415 阅读 · 2 评论 -
yolov5 focal_loss源码解析
yolov5 focal loss原创 2022-11-18 15:56:33 · 2215 阅读 · 0 评论 -
Pytorch nn.BCEWithLogitsLoss()和nn.BCELoss()
nn.BCELoss()原创 2022-11-18 10:53:10 · 636 阅读 · 0 评论 -
torch保存和加载模型的两种方式
# 保存整个网络torch.save(net, PATH) # 保存网络中的参数, 速度快,占空间少torch.save(net.state_dict(),PATH)#--------------------------------------------------#针对上面一般的保存方法,加载的方法分别是:model_dict=torch.load(PATH)model_dict=model.load_state_dict(torch.load(PATH))...原创 2022-03-23 16:00:22 · 2818 阅读 · 0 评论 -
pytorch多gpu训练,单机多卡,多机多卡
pytorch多gpu训练,单机多卡,多机多卡_kejizuiqianfang的博客-CSDN博客_pytorch多机多卡转载 2021-09-14 09:30:44 · 532 阅读 · 0 评论 -
gan网络原理(通俗)+minist手写字体实战
gan网络原理如下:mnist手写字体实战:import torchimport torchvisionfrom torchvision import transformsfrom torchvision.utils import save_imagefrom torch import nnfrom torch.autograd import Variablefrom torch import optimimport ostransform = transforms.C.原创 2021-07-29 19:56:30 · 453 阅读 · 0 评论 -
pytorch 使用amp.autocast半精度加速训练
https://blog.csdn.net/weixin_42216799/article/details/110876374转载 2021-06-22 13:58:43 · 466 阅读 · 0 评论 -
样本不均衡---权重设置(keras,sklearn)class_weight
https://www.cnblogs.com/qi-yuan-008/p/11992156.htmlhttps://blog.csdn.net/weixin_38385446/article/details/88948076转载 2020-07-19 16:32:15 · 6063 阅读 · 0 评论 -
keras中添加正则化
正则项正则项在优化过程中层的参数或层的激活值添加惩罚项,这些惩罚项将与损失函数一起作为网络的最终优化目标惩罚项基于层进行惩罚,目前惩罚项的接口与层有关,但Dense, Conv1D, Conv2D, Conv3D具有共同的接口。这些层有三个关键字参数以施加正则项: kernel_regularizer:施加在权重上的正则项,为...原创 2020-07-19 16:44:42 · 1475 阅读 · 0 评论 -
python各种镜像安装包下载链接,附torch
阿里云 http://mirrors.aliyun.com/pypi/simple/中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/豆瓣(douban) http://pypi.douban.com/simple/清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/使用案例:https://pyp..转载 2020-07-28 16:30:27 · 2623 阅读 · 0 评论 -
torchversion模块介绍
torchversion模块介绍https://blog.csdn.net/my_kingdom/article/details/85201393torchversion.models 包含常用的分类、检测和分割模型:例如导入分类模型:from torchversion.models import resnet152,vgg19_bn导入检测模型:from torchversion.models.detection import mask_rcnn,faster-rcnn,rpn原创 2020-09-11 18:03:15 · 4158 阅读 · 1 评论 -
keras保存和加载模型的方式
keras保存模型的两种方式:model.save("m1.h5") # 既保存了模型的图结构,又保存了模型的参数model.save_weights("m2.h5") #只保存了模型的参数keras加载模型的两种方式:#第一种(通用性网络可用)from keras.models import load_modelmodel = load_model("weight.h5")#第二种(适用于有创新的网络,例如自创的激活函数等)from Nets.mobilenetV3_s原创 2020-07-30 09:27:57 · 483 阅读 · 0 评论 -
keras和tensortflow对应版本
https://www.cnblogs.com/carle-09/p/11661261.html转载 2020-04-30 14:18:13 · 500 阅读 · 0 评论