Giters

halbielee / EPS

Official PyTorch implementation of "Railroad is not a Train: Saliency as Pseudo-pixel Supervision for Weakly Supervised Semantic Segmentation", CVPR2021

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Train EPS on other dataset.

JimmyMa99 opened this issue · comments

commented

I want to know about how to train EPS on coco17.
I think I should get the cls_labels.npy in the file metadata/coco.
Can you tell me how to transfer the official json to the .npy?

commented

Hello, @JimmyMa99

If you want to use another dataset (e.g., ms-coco 2017), you should know the labels for each image and make the class labels into binary vector. So, you have to get 'label list' from the json file and convert it to ".npy" file.

The official cocodataset provides how to handle the json file. Please check the official github.

commented

Thanks, @halbielee

I have written a tool to read the json of the coco17 dataset according to the format, maybe other researchers who wish to use your project can go to the following link to set up cls_labels.npy for the coco17 dataset.
https://github.com/JimmyMa99/coco17-get-cls_label/tree/main

commented

How nice you are, @JimmyMa99!

I think what you did can help other researchers as you said.
I will post your way to set up "cls_labels.npy" for coco17 dataset on the README file!

Thank you!