PMRID-Pytorch

this is a project for training and testing PMRID

some code copy from megvii-research/PMRID

generate_list.py is for generate list for trainning

main.py is for training and testing dataset