def __getitem__()

in utils/data_loader.py [0:0]

• 24 lines of code
• 1 McCabe index (conditional complexity)

	def __getitem__(self, index):
		imroot = self.imroot
		hroot = self.hroot
		oproot = self.oproot
		vocab = self.vocab
		annotation = self.annotation
		path, end = annotation.anns[index]
		images = []
		poses = []
		poses2 = []
		homography = []
		for i in range (end-self.seq_length, end):
			image, upp_pose, low_pose, h, pose2 = getPair(imroot, hroot, oproot, path, vocab, i)
			if self.transform is not None:
				image = self.transform(image)
			poses.append((upp_pose, low_pose))
			images.append(image)
			homography.append(h)
			poses2.append(pose2)
		homography = torch.Tensor(homography)
		images = torch.stack(images)
		target = torch.Tensor(poses)
		poses2 = torch.Tensor(poses2)
		return images, target, homography, poses2