pytorch
北斗星辰001
用不抱怨的生活方式去追求梦想,去实现自己的人生价值,好运将时刻伴随,奋斗拼搏,绝不碌碌无为的过一生!
展开
-
pytorch语义分割中CrossEntropyLoss()损失函数的理解与分析
最近在尝试使用pytorch深度学习框架实现语义分割任务,在进行loss计算时,总是遇到各种问题,针对CrossEntropyLoss()损失函数的理解与分析记录如下:1.数据准备为了便于理解,假设输入图像分辨率为2x2的RGB格式图像,网络模型需要分割的类别为2类,比如行人和背景。训练的时候,网络输入图像的shape为(1,3,2,2)。网络的输出预测图像的shape为(...原创 2020-03-28 15:30:20 · 15252 阅读 · 38 评论 -
pytorch官方语义分割教程报错
在尝试复现pytorch官方语义分割教程时出现以下问题:完整的错误如下:Traceback (most recent call last): File "/home/uzi/anaconda3/envs/pytorch-py3/lib/python3.5/site-packages/numpy/core/function_base.py", line 118, in linspace...原创 2020-03-21 15:25:36 · 1062 阅读 · 4 评论