语义分割训练时 Keras 损失为 NaN
我正在使用头部分割数据集。单个掩码看起来像这样 所有遮罩图像都是单个通道。这是我的代码： image_size = 512 batch = 4 labels = 14 data_director…
如何在pytorch中交换x和y？
给定一个输出张量为 100 * 6(xmin,ymin,xmax,ymax,conf,class)，如何在 pytorch 中得到一个 100 * 6(ymin,xmin,ymax,xmax,conf,class) 的张量？ 例如…
如何为多输出回归问题编写 Pytorch Dataset 类和 DataLoader？
我正在处理基于事件的数据（包括时间戳、x 和 y 坐标以及极性）来解决神经形态问题。但是，我的问题是，当我尝试预测多个输出时，如何创建数据集和正…
“无类型” Pytorch 模型测试部分中的对象不可迭代？
我正在使用 Pytorch 模型研究高光谱图像。当我想测试模型时，出现“NoneType objectis not iterable”错误。图像上有一个滑动窗口，我将其添加到测试…
Pytorch dist.all_gather_object 挂起
我正在使用 dist.all_gather_object （PyTorch 版本 1.8）从所有 GPU 收集样本 ID： for batch in dataloader: video_sns = batch["video_ids"] logit…
为什么我的模型无法学习？损失非常高
我建立了一个模拟模型，卡车根据垃圾容器的填充水平收集垃圾容器。 我使用 OpenAi Gym 和 Tensorflow/keras 来创建我的深度强化学习模型...... 但是我…
使用 matplotlib 按类别着色
这是我用 coco json 和图像创建带注释的图像的代码。 I = io.imread('%s/images/%s'%(dataDir,img['file_name'])) axes = plt.gca() axes.set_frame_o…
如何从自定义神经网络模型中获取逻辑和概率
以下源代码可以从 Tensorflow 中的 imagenet 预训练模型中获取概率和 logits import tensorflow as tf import tensorflow.contrib.slim as slim impor…
尽管加载了模型的权重，为什么模型摘要中的不可训练参数为零？
使用该命令 torch.save(model.state_dict(), 'model.pth') 我在训练模型后 保存参数。 但是，当我使用命令 model = EfficientNetModel() MODEL_PATH =…
损失率不错，但模型精度始终为 0
我正在使用 fastai 处理事故数据集，损失率随着每个时期的推移而不断下降，但准确性保持在 0.000000。 下面是我的 google colab 笔记本。 https://col…
