(self,
pixel_grading_num: int = 100,
frame_grading_num: int = 13)
| 407 | flop = FlopEstimator2D1DNotExact |
| 408 | |
| 409 | def __init__(self, |
| 410 | pixel_grading_num: int = 100, |
| 411 | frame_grading_num: int = 13) -> None: |
| 412 | super().__init__() |
| 413 | self.pixel_grading_arr = np.linspace( |
| 414 | self.side_list[0]**2, |
| 415 | self.side_list[-1]**2, |
| 416 | pixel_grading_num, |
| 417 | dtype=np.int64) |
| 418 | self.frame_grading_arr = np.arange(frame_grading_num) * 4 + 1 |
| 419 | |
| 420 | def __call__(self, n_frame: int, height: int, width: int, |
| 421 | rnd_state: np.random.RandomState) -> Any: |