MultiBox_Loss bug改进

最新推荐文章于 2022-04-16 13:12:23 发布

AI算法网奇

最新推荐文章于 2022-04-16 13:12:23 发布

阅读量1.8k

点赞数

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/118217228

版权

pytorch知识宝典专栏收录该内容

499 篇文章 227 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

这篇博客指出网上常见的MultiBoxLoss实现存在bug，可能导致训练时崩溃。作者提供了正确的代码示例，并详细说明了需要修改的部分，帮助读者避免遇到相同问题。

摘要由CSDN通过智能技术生成

我发现网上大多数MultiBoxLoss的代码是有bug的，导致训练时不定时的崩溃

比如这个代码：

https://zhuanlan.zhihu.com/p/77868999

正确代码：

# -*- coding:utf-8 -*-

from __future__ import division
from __future__ import absolute_import
from __future__ import print_function
import configparser
import math
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable

from focal_loss import FocalLoss
from utils.box_utils import match, log_sum_exp
from data import cfg_mnet

GPU = cfg_mnet['gpu_train']


class MultiBoxLoss(nn.Module):
    """SSD Weighted Loss Function
    Compute Targets:
        1) Produce Confide

了解本专栏