according to eq 23, and the shift scale for 1024 is 4 here, not consistent with paper, but It's ok. Change of base_scale will make shift_scale of 256 != 1
(
self,
t,
frames,
height,
width,
)
| 64 | return t |
| 65 | |
| 66 | def sigma_shift( |
| 67 | self, |
| 68 | t, |
| 69 | frames, |
| 70 | height, |
| 71 | width, |
| 72 | ): |
| 73 | """according to eq 23, and the shift scale for 1024 is 4 here, not |
| 74 | consistent with paper, but It's ok. |
| 75 | |
| 76 | Change of base_scale will make shift_scale of 256 != 1 |
| 77 | """ |
| 78 | shift_scale = self.get_shift_scale(frames, height, width) |
| 79 | return self.sigma_shift_given_shift_scale(t, shift_scale) |
| 80 | |
| 81 | @staticmethod |
| 82 | def sigma_shift_given_shift_scale(t, shift_scale): |
no test coverage detected