(self)
| 62 | self.current_epoch = 0 |
| 63 | |
| 64 | def _register_videos(self): |
| 65 | for label_name in self.label_files: |
| 66 | video_name = '/'.join(label_name.split('/')[:-1]) |
| 67 | if video_name not in self.video_dict: |
| 68 | print("register {}-th video: {} ".format(len(self.video_dict) + 1, video_name)) |
| 69 | self.video_dict[video_name] = len(self.video_dict) |
| 70 | # assert len(self.video_dict) <= 300 |
| 71 | |
| 72 | def set_epoch(self, epoch): |
| 73 | self.current_epoch = epoch |