Skip to content
This repository was archived by the owner on Jul 2, 2021. It is now read-only.

Add training part of PSPNet #634

Merged
merged 42 commits into from
Feb 6, 2019
Merged

Add training part of PSPNet #634

merged 42 commits into from
Feb 6, 2019

Conversation

yuyu2172
Copy link
Member

@yuyu2172 yuyu2172 commented Jun 1, 2018 •
edited
Loading

Merge after #749
Merge after #743

@yuyu2172 yuyu2172 changed the title Add training part of PSPNet [WIP] Add training part of PSPNet Jun 2, 2018
@yuyu2172
Copy link
Member Author

yuyu2172 commented Oct 24, 2018 •
edited
Loading

I pretrained PSP's feature extractor with ImageNet.

top 1 error: 21.45

code I used
https://github.com/yuyu2172/chainercv/blob/resnet-train-dev/examples/classification/model.py

weights
https://drive.google.com/file/d/184hZjmsT7XUmGqrd6lS9HMSbGhE_1Xog/view?usp=sharing

@yuyu2172
Copy link
Member Author

yuyu2172 commented Dec 5, 2018

Bugs I found

@yuyu2172 yuyu2172 mentioned this pull request Dec 5, 2018
Closed
3 tasks
@yuyu2172
Copy link
Member Author

yuyu2172 commented Dec 8, 2018 •
edited
Loading

Goal: 77.9 (https://github.com/holyseven/PSPNet-TF-Reproduce)

Cityscapes: mIoU 0.7391

8f72cb7 mIoU: 0.7515

1e9953d (n_iter 30000) mIou: 0.7537
input_size = 720, n_iter=90000 mIoU: 0.7572

@yuyu2172 yuyu2172 added this to the 0.12 milestone Dec 19, 2018
@yuyu2172 yuyu2172 changed the title [WIP] Add training part of PSPNet Add training part of PSPNet Dec 19, 2018
Hakuyume
Hakuyume reviewed Feb 6, 2019
View reviewed changes
chainercv/experimental/links/model/pspnet/pspnet.py
'ade20k': {
'param': {'n_class': 150, 'input_size': (473, 473)},
'url': 'https://chainercv-models.preferred.jp/'
'pspnet_resnet101_ade20k_trained_2018_12_23.npz',
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

How about updating this block?

chainercv/chainercv/experimental/links/model/pspnet/pspnet.py

Lines 131 to 138 in f43f891

* :obj:`'cityscapes'`: Load weights trained on the train split of \
Cityscapes dataset. \
:obj:`n_class` must be :obj:`19` or :obj:`None`.
* :obj:`'imagenet'`: Load ImageNet pretrained weights for \
the extractor.
* `filepath`: A path of npz file. In this case, :obj:`n_class` \
must be specified properly.
* :obj:`None`: Do not load weights.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

got it

@Hakuyume Hakuyume mentioned this pull request Feb 6, 2019
Merged
@Hakuyume Hakuyume merged commit 9629c30 into chainer:master Feb 6, 2019
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Reviewers

@Hakuyume Hakuyume Hakuyume approved these changes

Assignees

@Hakuyume Hakuyume

Labels
feature
Projects
None yet
Milestone
0.12
Development

Successfully merging this pull request may close these issues.
3 participants
@yuyu2172 @Hakuyume @knorth55