Skip to content

Training a new model #23

@sidc9

Description

@sidc9

Hi,

Thank you for the code! It has been very useful to me. But I have run into a snag and I was hoping you can help.

I have been trying to train my own model with your code. I have a dataset of around 65000 images with 49 landmark points for each image.

While setting the parameters, I set "global_params.landmark_num" in InitializeGlobalParam() to 49. But the execution always crashes while loading the images at around the 16000 image mark or 30000 mark with the error - OpenCV Out of Memory exception.

However, if I leave the landmark_num to 68 the training completes successfully. But then if I try to Test the model or run the Demo mode, the program always crashes (which I think is what should happen since the global parameter claims there are 68 points but in reality there are only 49).

It would be really good if you could help me out a little.

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