Skip to content

训练数据集尺寸的问题 #12

@zkxwy1996

Description

@zkxwy1996

大佬您好,我想向您请教一下,我用您的模型训练 人头检测:
(1)我用的数据集图片尺寸大部分在500左右×1500左右,这样的话在使用您的模型的时候需要调整图片resize的尺寸嘛?还是直接用您模型里面的640*640就好呢?图片输入的尺寸对于整体训练会有影响吗?
(2)原论文里有对数据集的人头尺寸的分布做处理,就是尽量保证各个尺寸的人头的数量保持在1:1:1:1:1:1:1:1,我的数据集的人头基本上都是一个尺寸的,这样的话可以直接用您的模型吗?还是说需要做些什么处理?
(3)我的cls_loss降到三左右就降不下去了,然后用训练的参数predict的时候检测不出来框,请问有什么好的解决办法嘛?
谢谢您!
再次向您的开源致敬!

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