Paddle Focal loss怎么设置?
收藏
Focal Loss 被提出用于解决计算机视觉任务中前景-背景不平衡的问题。该OP先计算输入x中每个元素的sigmoid值,然后计算sigmoid值与类别目标值label之间的Focal Loss。
但API中fgnum不知如何提供,这是API解释:
fg_num (Variable) – 维度为 [1][1] 的1-D Tensor,表示批量内正样本的数量,需在进入此OP前获取正样本的数量。数据类型为int32。
caltech101数据集,假如训练批数量16,random抽取图片,请问这里fg_num如何设置?
1
收藏
请登录后评论
顶
请问这里fg_num如何设置?
API没有提供fg_num这个参数
你可以到源代码里面看看 能不能找到这个参数
取出label里面>1的个数就行了
每批次16个样本,样本是random取出的,类别数也是16个,这时候怎么设置?
有这个的,sigmoid focal loss ,我再paddle官方api搜到的
没人用过吗
没用过,你看这样行不行。一个batch随机抽M张图片,每张图片有N个ground_truth,然后看算法是怎样计算正样本的咯,例如:将与gt有最大IoU的anchor划分为TP,fg_num+=1.最后将一个batch内的所有正样本加起来,就可以传给focal loss了。
可以先试试看