Skip to content

这里给大家提几个小建议! #83

@xs-web-lhdd

Description

@xs-web-lhdd

1、作者提供的网盘数据集中JPEGImages要检查所有图片后缀都是jpg,把JGP的改为jpg要不然会在数据转换时运行报错
2、在yolo.py里面

        for mi, s in zip(m.m, m.stride):  #  from
            b = mi.bias.view(m.na, -1)  # conv.bias(255) to (3,85)
            with torch.no_grad(): # 在原有代码基础上添加这行代码
                b[:, 4] += math.log(8 / (640 / s) ** 2)  # obj (8 objects per 640 image) # 缩进
                b[:, 5:] += math.log(0.6 / (m.nc - 0.99)) if cf is None else torch.log(cf / cf.sum())  # cls # 缩进
            mi.bias = torch.nn.Parameter(b.view(-1), requires_grad=True)

3、在output_to_target中进行修改,将以下行:
return np.array(targets)修改为:return np.array(torch.tensor(targets).cpu())

以上代表我个人观点,大佬轻喷哈哈哈哈哈!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions