Skip to content

Pb with number of classed predicted #18

Description

@BenLag2906

Hi,
When I started to learn less class with SSD ( 2 classes) : it diverge quickly.

Should I intiailizing my CNN differently ? beacuse my pretrained is based on 20 classes...

Or

Is there a modification to add with my softmax function ?

Could you please give a piece of advise ?
regards

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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