Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
kiitosu opened this issue May 16, 2019 · 3 comments

Comments

@kiitosu
Copy link

kiitosu commented May 16, 2019

No description provided.

@kiitosu kiitosu 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 May 16, 2019
@kiitosu kiitosu 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 May 16, 2019
@kiitosu
Copy link
Author

kiitosu commented May 16, 2019

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?

@mistaker-likf
Copy link

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

@kiitosu
Copy link
Author

kiitosu commented Jan 16, 2020

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
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants