train code + inference code were released. [pytorch version]
first enter the directory: and follow the instruction: 1.write the dataset path in mypath.py 2. and run
python train.py
after trainning,it will save two model: convnet.pth,pan.pth
#run:
python inference.py
