(self, txt_path, width, height, n_sample_frames, sample_frame_rate, enable_inpaint=True, face_aligner=None)
| 102 | |
| 103 | class LargeScalePortraitVideos(Dataset): |
| 104 | def __init__(self, txt_path, width, height, n_sample_frames, sample_frame_rate, enable_inpaint=True, face_aligner=None): |
| 105 | self.txt_path = txt_path |
| 106 | self.width = width |
| 107 | self.height = height |
| 108 | self.n_sample_frames = n_sample_frames |
| 109 | self.sample_frame_rate = sample_frame_rate |
| 110 | self.enable_inpaint = enable_inpaint |
| 111 | self.video_files = self._read_txt_file_images() |
| 112 | self.face_aligner = face_aligner |
| 113 | |
| 114 | def _read_txt_file_images(self): |
| 115 | with open(self.txt_path, 'r') as file: |
nothing calls this directly
no test coverage detected