提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
lovasz-softmax模块安装
前言
提示:这里可以添加本文要记录的大概内容:
最近在搞那个unet++的复现,里面有个损失函数模块需要from LovaszSoftmax.pytorch.lovasz_losses import lovasz_hinge,我当时就不知道,然后去网上到处找,什么妖魔鬼怪都来了,什么pip,conda,我一直安装不了,给我人都整麻了,然后我就去找源码,突然发现应该是这样的:
提示:以下是本篇文章正文内容,下面案例可供参考
直接进这里
https://github.com/bermanmaxim/LovaszSoftmax/tree/master/pytorch
然后直接下载 lovasz_losses.py
最后from lovasz_losses import lovasz_hinge