PyTorch2 [TDAN] Tdan: Temporally-deformable alignment network for video super-resolution 논문 요약 이전의 VSR methods는 보통 reference frame과 supporting frame사이의 optical flow를 구한 후 이를 사용해 supporting frame을 warp하여 temporal alignment를 한다. 하지만 부정확한 flow와 iamge-level warping은 warped supporting frames에서 부적절한 artifacts를 유발한다. 이를 해결하기 위한 temporal-deformable alignment network (TDAN)은 optical flow를 사용하지 않고 reference frame과 supporting frame을 feature level에서 적응형으로 align한다. 이는 refrence frame과 supporting .. 2022. 9. 14. [Pytorch] Pytorch로 GPU 사용 가능한지 확인하기 Pytorch를 통해 GPU를 사용할 수 있는지, 그 수는 몇인지 확인하는 방법 import torch # CUDA 사용한지 여부 확인: True or False print(torch.cuda.is_available()) # GPU 수 확인 print(torch.cuda.device_count()) # 현재 사용하는 GPU 번호 확인 print(torch.cuda.current_device()) # 0번 GPU 이름 확인: ex) GeForce GTX 950M print(torch.cuda.get_device_name(0)) Reference https://stackoverflow.com/questions/48152674/how-to-check-if-pytorch-is-using-the-gpu How t.. 2022. 5. 15. 이전 1 다음