neulus/cloud-removal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pixel shuffle

Browse Source
This commit is contained in:
neulus
2025-10-01 16:30:05 +09:00
parent 8966cafb8f
commit 49025c4d87
19 changed files with 162 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
quick_eval.py
View File

@@ -10,14 +10,14 @@ from src.dataset.preprocess import denormalize
from src.model.utransformer import UTransformer
from src.rf import RF
checkpoint_path = "artifact/wild-wave-3/checkpoint_epoch_100.pt"
device = "cuda:0"
checkpoint_path = "artifact/icy-field-12/checkpoint_epoch_260.pt"
device = "cuda:2"
save_dir = "test_images"
os.makedirs(save_dir, exist_ok=True)
model = UTransformer.from_pretrained_backbone(
"facebook/dinov3-vits16-pretrain-lvd1689m"
"facebook/dinov3-vitl16-pretrain-sat493m"
).to(device)
checkpoint = torch.load(checkpoint_path, map_location=device)
@@ -28,7 +28,7 @@ rf.model.eval()
_, test_dataset = get_dataset()
batch_size = 32
batch_size = 1
psnr_sum = 0
ssim_sum = 0
lpips_sum = 0