我目前正在為一個(gè)項(xiàng)目做一些 AI 研究,為此我必須習(xí)慣一個(gè)名為“Pytorch”的框架。這很好,除了按照官方教程(在此處找到)之外,代碼無(wú)法正常運(yùn)行。這個(gè)想法是我從準(zhǔn)備好的數(shù)據(jù)集中分析一組面部特征,然后用它做一些事情(還沒(méi)有達(dá)到那個(gè)部分)。但是當(dāng)我運(yùn)行這段代碼時(shí):img_name = os.path.join(self.root_dir, self.landmarks_frame.iloc([index, 0])) # At this point 'index' is 0數(shù)據(jù)集初始化如下:face_dataset = fDataset(csv_file='faces/face_landmarks.csv', root_dir='faces/')這是彈出錯(cuò)誤的地方:for i in range(len(face_dataset)): sample = face_dataset[i] # <-- right there這導(dǎo)致了 getter 函數(shù):def __getitem__(self, index): img_name = os.path.join(self.root_dir, self.landmarks_frame.iloc([index, 0])) image = io.imread(img_name) landmarks = self.landmarks_frame.iloc[index, 1:].as_matrix() landmarks = landmarks.astype('float').reshape(-1, 2) sample = {'image': image, 'landmarks': landmarks}在我的FaceLandmarksDataset(Dataset):課堂上發(fā)現(xiàn)我只是得到了標(biāo)題的錯(cuò)誤。我發(fā)現(xiàn)這很奇怪,因?yàn)槲铱梢栽?PyCharm 中將數(shù)據(jù)集作為框架讀?。旱谝粡垐D清晰可見(jiàn)的地方。我也檢查過(guò)它是否在我正在查看的文件夾中。任何人都可以幫忙嗎?:)
添加回答
舉報(bào)
0/150
提交
取消