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

Segmentation maps for real dataset #135

Open
melisandeteng opened this issue May 14, 2020 · 3 comments
Open

Segmentation maps for real dataset #135

melisandeteng opened this issue May 14, 2020 · 3 comments
Assignees

Comments

@melisandeteng
Copy link
Contributor

We need to save segmentation maps of the real dataset (not only categories ground/background, but with the full set of labels vegetation, cars, etc.).
A version with 19 Cityscapes labels will be computed, and we will also make a version of the segmentation maps so that the labels match our simulated dataset.
You can check the correspondence between Cityscapes and our simulated world segmentation maps labels here.

@melisandeteng melisandeteng self-assigned this May 14, 2020
@melisandeteng
Copy link
Contributor Author

melisandeteng commented May 14, 2020

We used Deeplabv3+MobileNet model to generate segmentation maps for our streetview dataset of 1950 images available at /network/tmp1/ccai/data/munit_dataset/non_flooded/streetview_mvp.
This is a reduced version of our real dataset on which depth maps were computed using MegaDepth - it does not contain all the images of the real dataset because MegaDepth requires certain input sizes and we did not want to squish images to be able to use MegaDepth.

The obtained images are available here :
/network/tmp1/ccai/data/munit_dataset/non_flooded/streetview_mvp_deeplabv3/

xxx_image.png is the original square image resized to 512*512
xxx_pred.png is the segmentation map. Note that is does not show with colors but it is encoded with the label ids [0-18]

@melisandeteng
Copy link
Contributor Author

It seems that Deeplabv3 does not do such a great job at segmenting sky on Streetview images, especially when there are clouds (there is a lot of warping in the sky) (see example below) .
So we will try the HRNet + OCR + SegFix model which is SOTA on Cityscapes
image

@melisandeteng
Copy link
Contributor Author

Segmentation maps computed with HRNet model are available at
/network/tmp1/ccai/data/munit_dataset/non_flooded/streetview_mvp_seg_HRNetfor the cityscapes labels and /network/tmp1/ccai/data/munit_dataset/non_flooded/streetview_mvp_seg_HRNet_simlabelsfor the labels (merged from cityscapes) that correspond to our simulated dataset labels.

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

1 participant