| 32 | ('HD_dataset/HD544p_GT/Sintel_Temple2_1280x544.yuv', 544, 1280), |
| 33 | ] |
| 34 | def inference(I0, I1, pad, multi=2, arbitrary=True): |
| 35 | img = [I0, I1] |
| 36 | if not arbitrary: |
| 37 | for i in range(multi): |
| 38 | res = [I0] |
| 39 | for j in range(len(img) - 1): |
| 40 | res.append(model.inference(img[j], img[j + 1])) |
| 41 | res.append(img[j + 1]) |
| 42 | img = res |
| 43 | else: |
| 44 | img = [I0] |
| 45 | p = 2**multi |
| 46 | for i in range(p-1): |
| 47 | img.append(model.inference(I0, I1, timestep=(i+1)*(1./p))) |
| 48 | img.append(I1) |
| 49 | for i in range(len(img)): |
| 50 | img[i] = img[i][0][:, pad: -pad] |
| 51 | return img[1: -1] |
| 52 | |
| 53 | tot = [] |
| 54 | for data in name_list: |