pytorch-PCN
pytorch-PCN copied to clipboard
siriusdemon
Pytorch implementation of PCN
This is the loop for your stage1 which invokes first model repeatedly: ``` def stage1(img, imgPad, net, thres): row = (imgPad.shape[0] - img.shape[0]) // 2 col = (imgPad.shape[1] - img.shape[1])...
Hello. I tested it on the GPU, and it slowed down a lot. After debugging, I found that pcn.py→def stage1(),These two double "for" take a lot of time. for i...
I try the demo, found that it uses CPU only, how can I use GPU with the model?
> root@ada39ed4587c:/app/pytorch-PCN# pip3 install . WARNING: pip is being invoked by an old script wrapper. This will fail in a future version of pip. Please see https://github.com/pypa/pip/issues/5599 for advice on...
hi what changes in code required to check several numbers of images output in one go. not only one image
Hi, nice work. How about retraining on a new dataset? Because I find that the pre-trained PCN performs badly on low-resolution images.
I compare the origin repo with yours. function ./pcn/pcn.py/stage1(img, imgPad, net, thres) line164.`img_resized` is changed by function `preprocess_img` in each loop. Is it correct? I don't know whether the operator...
