(self, path_vid)
| 465 | class VGGDataset(Dataset): |
| 466 | |
| 467 | def __init__(self, path_vid): |
| 468 | self.data_complete = pd.read_json('./data/data.json',orient='records',dtype=False,lines=True) |
| 469 | |
| 470 | self.vid = [] |
| 471 | with open('./data/vids/'+path_vid, "r") as fr: |
| 472 | for line in fr.readlines(): |
| 473 | self.vid.append(line.strip()) |
| 474 | self.data = self.data_complete[self.data_complete.video_id.isin(self.vid)] |
| 475 | self.data['video_id'] = self.data['video_id'].astype('category') |
| 476 | self.data['video_id'].cat.set_categories(self.vid, inplace=True) |
| 477 | self.data.sort_values('video_id', ascending=True, inplace=True) |
| 478 | self.data.reset_index(inplace=True) |
| 479 | |
| 480 | self.framefeapath='./data/ptvgg19_frames/' |
| 481 | |
| 482 | |
| 483 | def __len__(self): |
nothing calls this directly
no outgoing calls
no test coverage detected