Skip to content

inference result with model trained with train.py is not good and model size is almost twise bigger then pretrained model #62

Open

Description

@kiitosu
No description provided.

Activity

  1. changed the title [-]Train result not good and,[/-] [+]inference result with netword trained with train.py is not good and model size is almost twise bigger then pretrained model[/+] on May 16, 2019
  2. changed the title [-]inference result with netword trained with train.py is not good and model size is almost twise bigger then pretrained model[/-] [+]inference result with model trained with train.py is not good and model size is almost twise bigger then pretrained model[/+] on May 16, 2019
  3. kiitosu commented on May 16, 2019

    @kiitosu
    Author

    I tried to train PSPNet101 with cityscapes dataset.
    I did below.
    ・download dataset
    ・edit cityscapes_trail_list.txt gtFine_labelTrainIds.png to gtFine_labeIds.png
    ・edit train.py DATA_DIRECTORY to suit my image data
    ・change batch size 2 to 1
    ・python train.py

    After 15hours I finished the training .
    Inference with inference.py is so poor.

    Traind data I created is
    File size of model.ckpt-60000.data-00000-of-00001 is about 502M byte.
    File size of model.ckpt-60000.meta is about 2.7M byte.

    Pretraind data is
    File size of model.ckpt-0.data-00000-of-00001 size is 254M .
    File size of model.ckpt-0.meta size is 254M .

    I think the problem is the model trained is not proper because of the model.ckpt-60000.data-00000-of-00001 size is twice bigger than pretrained file model.ckpt-0.data-00000-of-00001.

    Or the problem is model.ckpt-60000.meta size is too small compared wit pretrained file model.ckpt-0.meta size is 254M.

    How could I solve this problem?

  4. mistaker-likf commented on Jan 15, 2020

    @mistaker-likf

    My problem is same with you. Have you solve the issue?

  5. kiitosu commented on Jan 16, 2020

    @kiitosu
    Author

    I'm sorry. No I haven't.

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