Skip to content

How to train with my own dataset? #34

Open
@knwng

Description

Hi, thanks for sharing this excellent project. I've successfully trained it on VOC dataset and got pretty good result and speed, but when I try to train it using my own dataset(with only 1 class). It always get 0 mAP and nan loss as testing.
I've already change the num_class in detect_loss layer and associated params in reg_reshape and the last conv layer and I've pre-proccessed my input data(reshape and subtract mean value).
What confuse me is the meaning of the other params in detect_loss layer like coord_scale, noobject_scale...Should I change them for new dataset? If so, how should I calculate them? Thanks in advance!

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