Functions — github.com/cxzhou95/XLSR

CodeHub MCPcopy Create free account

hub / github.com/cxzhou95/XLSR / functions

Functions31 in github.com/cxzhou95/XLSR

⨍ Functions 31 ◇ Types & classes 8

↓ 7 callersFunctioncreate_dataloader

(split, SR_rate, augment, batch_size=1, shuffle=False, num_workers=1, pin_memory=True)

↓ 5 callersMethod__init__

(self, SR_rate)

↓ 4 callersFunctioncal_psnr

Parameters ---------- x, y are two tensors has the same shape (1, C, H, W) Returns ------- score : PSNR.

↓ 3 callersFunctionsave_res

Parameters ---------- preds : List each pred has a shape of 1x3xHxW. BGR img_names : List Returns ------- None.

visualization.py:16

↓ 1 callersMethodfuse_model

↓ 1 callersFunctionmeasure_speed

(model, device)

↓ 1 callersFunctionrandom_crop

(LR_img, HR_img, crop_size, SR_rate)

↓ 1 callersFunctiontest

(model, dataloader, device, txt_path)

↓ 1 callersFunctiontest

(model, dataloader, device, txt_path)

↓ 1 callersFunctiontrain

(model, dataloader, criteria, device, optimizer, scheduler)

↓ 1 callersFunctiontrain_one_epoch

(model, criterion, optimizer, train_dataloader, device='cpu', scheduler=None)

↓ 1 callersFunctionvalidation

(model, dataloader, criteria, device)

↓ 1 callersFunctionvalidation

(model, dataloader, criteria, device='cpu')

↓ 1 callersFunctionvisualize_training

visualization.py:35

Method__getitem__

(self, eps=0.01)

(self, split, SR_rate, augment=False)

(self, in_channels, out_channels, kernel_size, bias=True)

(self, in_channels, out_channels, kernel_size=3, groups=4)

(self, in_channels, out_channels, groups)

(self, SR_rate)

(self, pred, gt)

Functionprint_size_of_model