-
Notifications
You must be signed in to change notification settings - Fork 118
Description
> > > I write a python script to prepare the data, maybe you want to try it
https://gist.github.com/qzane/4d07b7551914f97f2bf8b9c79138ab14
@qzane Thank you for your python script. But this only generates the edges file and the respective blank(black) images. Do you have a python script to generate labels as well?
@gayalkuruppu I don't think you need labels to get the correct output. I created a folder with custom images, ran the python script that was linked to in a previous comment (https://gist.github.com/qzane/4d07b7551914f97f2bf8b9c79138ab14 and I saved it as prepare_dataset.py) with
python prepare_dataset.py custom_dataset/CIHP/images CIHPand it created a new folder within datasets/ called CIHP (it'll be named whatever you use as the last argument in the previous python command).
The new folder has the structure
CIHP_PGN/ ├─ datasets/ │ ├─ CIHP/ │ │ ├─ edges/ │ │ │ ├─ image0.png │ │ │ ├─ image1.png │ │ ├─ images/ │ │ │ ├─ image0.png │ │ │ ├─ image1.jpg │ │ ├─ list/ │ │ │ ├─ val_id.txt │ │ │ ├─ val.txtand then I was able to run
python test_pgn.pyand see the results in CIHP_PGN/output/cihp_parsing_maps/
Sorry to bother you now, I just tried to run this script, but the resulting edges are full black images and not producing the correct edge images. How to solve this please? Thank you~
Originally posted by @l-mengwei in #38 (comment)