NotificationsYou must be signed in to change notification settings
Fork371
Star1.7k

Commit9f2fa4a

committed

Removed shape loading which fixes a bug with loading AFLW2000 and 300W-LP for training and testing.

1 parent888dc77 commit9f2fa4aCopy full SHA for 9f2fa4a

File tree

1 file changed

-10

lines changed

code
- datasets.py

1 file changed

-10

lines changed

`‎code/datasets.py‎`

Lines changed: 2 additions & 10 deletions

Original file line number	Diff line number	Diff line change
`@@ -96,7 +96,6 @@ def __getitem__(self, index):`
`96`	`96`	`img=Image.open(os.path.join(self.data_dir,self.X_train[index]+self.img_ext))`
`97`	`97`	`img=img.convert(self.image_mode)`
`98`	`98`	`mat_path=os.path.join(self.data_dir,self.y_train[index]+self.annot_ext)`
`99`		`-shape_path=os.path.join(self.data_dir,self.y_train[index]+'_shape.npy')`
`100`	`99`
`101`	`100`	`# Crop the face loosely`
`102`	`101`	`pt2d=utils.get_pt2d_from_mat(mat_path)`
`@@ -136,11 +135,8 @@ def __getitem__(self, index):`
`136`	`135`	`bins=np.array(range(-99,102,3))`
`137`	`136`	`binned_pose=np.digitize([yaw,pitch,roll],bins)-1`
`138`	`137`
`139`		`-# Get shape`
`140`		`-shape=np.load(shape_path)`
`141`		`-`
`142`	`138`	`# Get target tensors`
`143`		`-labels=torch.LongTensor(np.concatenate((binned_pose,shape),axis=0))`
	`139`	`+labels=binned_pose`
`144`	`140`	`cont_labels=torch.FloatTensor([yaw,pitch,roll])`
`145`	`141`
`146`	`142`	`ifself.transformisnotNone:`
`@@ -171,7 +167,6 @@ def __getitem__(self, index):`
`171`	`167`	`img=Image.open(os.path.join(self.data_dir,self.X_train[index]+self.img_ext))`
`172`	`168`	`img=img.convert(self.image_mode)`
`173`	`169`	`mat_path=os.path.join(self.data_dir,self.y_train[index]+self.annot_ext)`
`174`		`-shape_path=os.path.join(self.data_dir,self.y_train[index]+'_shape.npy')`
`175`	`170`
`176`	`171`	`# Crop the face loosely`
`177`	`172`	`pt2d=utils.get_pt2d_from_mat(mat_path)`
`@@ -214,11 +209,8 @@ def __getitem__(self, index):`
`214`	`209`	`bins=np.array(range(-99,102,3))`
`215`	`210`	`binned_pose=np.digitize([yaw,pitch,roll],bins)-1`
`216`	`211`
`217`		`-# Get shape`
`218`		`-shape=np.load(shape_path)`
`219`		`-`
`220`	`212`	`# Get target tensors`
`221`		`-labels=torch.LongTensor(np.concatenate((binned_pose,shape),axis=0))`
	`213`	`+labels=binned_pose`
`222`	`214`	`cont_labels=torch.FloatTensor([yaw,pitch,roll])`
`223`	`215`
`224`	`216`	`ifself.transformisnotNone:`

0 commit comments

Comments

(0)

Movatterモバイル変換

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit9f2fa4a

File tree

1 file changed

1 file changed

`‎code/datasets.py‎`

0 commit comments