본문 바로가기

Vision3

Super Resolution EDT 사용하기 Super resolution 모델 중 하나인 EDT를 사용한 내용을 정리한다. 1. 깃 repository 다운로드 및 package 설치 EDT 모델은 Python 3.7 버전 이상, pytorch 1.4 버전 이상에서 돌아간다. (적절하게 설치한다) git clone https://github.com/fenglinglwb/EDT.git 를 입력하여 git 다운로드 후, 다운받은 EDT 폴더에서 requirement 에 있는 package 들을 설치한다. pip install -r requirements.txt 2. test_sample.py 코드 수정 EDT 모델은 test_sample.py 코드를 돌려서 사용가능한데, git에 업로드된 코드에 오류가 있어 수정을 해야한다. nano test_sam.. 2022. 10. 8.
WSL2, Nvidia GPU, CUDA 환경에서 super-resolution 모델 (SwinIR) 돌려보기 시스템 사양 (환경) CPU - i7-10700F Memory - 32GB Mainboard - MSI, B460M GPU - Geforce RTX 3060 Ti WSL - Ubuntu 20.04 WSL2 환경에서 Nvidia gpu를 (3060ti) 사용하여 SwinIR 이라는 모델로 super-resolution을 돌려보았다. SwinIR은 이전에 리뷰했던 super-resolution 모델과 달리 transformer 기반이라 따로 시도해 보았다. 먼저 AMD GPU를 이용한 DirectML로 구현을 해보고자 하였으나, DirectML의 pytorch에 attention 관련 함수들이 아직 완벽하게 구현되지 않아 코드를 돌릴수 없었던 관계로 결국, CUDA를 사용하였다. https://github... 2022. 7. 29.
WSL2, AMD GPU, DirectML 환경에서 super-resolution 모델 (tecogan) 돌려보기 시스템 사양 (환경) CPU - 3700X Memory - 32GB Mainboard - Msi, MAG B550M MORTAR GPU - RX 6800 (reference) WSL - Ubuntu 20.04 super-resolution 모델 중에서 tecogan 이라는 모델을 구현해 보자 이전 BSRGAN 이 pytorch 기반이었다면, 이번 모델은 tensorflow 기반 모델이라서 따로 한번 더 정리하였다. 정식 깃헙은 아래와 같다. GitHub - thunil/TecoGAN: This repo contains source code and materials for the TEmporally COherent GAN SIGGRAPH project. GitHub - thunil/TecoGAN: This.. 2022. 7. 25.