A Pytorch implementation of VPGNet
We have stopped maintaining this repo, yet its not completely run-able. We would recommand you to turn to other pytorch-based implementations or fork/contribute by your own. Thanks!
Recent updates would be uploaded here in time line.
Generate the file list with 5:1:1 (train:test:val):
path/to/database:~$ python3 gen_label_v4.py
Generate the labeled images and groundtruth image:
# im parameter: 'i' for original image, 'm' for masked image
path/to/database:~$ python3 gen_image.py im