How should I define the dataset path?

[BugMaker2002]

The UBFC-rPPG dataset I download from the Internet like this:

So the data path should be "/UBFC_rPPG/dataset2", I run the code using this command:

python main.py --gpu 0 --epochs 50 --batch_size 12 --lr 1e-4
--n_dim 2048 --temperature 0.1
--dataset_name "ubfc-rppg"
--dataset_dir "/share2/data/zhouwenqing/UBFC_rPPG/dataset2"
--workers 4 --vid_frame 150 --clip_frame 30 --roi_list 0 1 2 3 4 5 6 --stride_list 1 2 3 4 5
--log_dir "./logs/ubfc/train"
--model_depth 18

But it made a mistake:

I have taken a closer look at the code of your UBFCDataset class, but I find that the data set path your code handles does not seem to be the original UBFC-rPPG data set. Did you make any changes to the original UBFC-rPPG dataset directory format?

[Dylan-H-Wang]

Hi,
You need to preprocess the original dataset before using UBFCDataset. Please check codes here

SLF-RPM/utils/utils.py

Lines 311 to 396 in 298b01f

	def extract_ubfc_dataset(dataset_path, des_path_root):
	if not os.path.exists(des_path_root):
	os.makedirs(des_path_root)
	start = time.time()
	subjects = os.listdir(dataset_path)
	subjects.sort()
	# Detect face landmarks model
	fa = face_alignment.FaceAlignment(
	face_alignment.LandmarksType._2D,
	device="cuda:1",
	flip_input=False,
	face_detector="blazeface",
	)
	# Subject
	for subject in tqdm(subjects, desc="Extract UBFC-rPPG Dataset"):
	subject_path = os.path.join(dataset_path, subject)
	video_file_path = None
	mata_data_file_path = None
	try:
	# File paths
	for file in os.listdir(subject_path):
	if file.endswith(".avi"):
	video_file_path = os.path.join(subject_path, file)
	elif file.endswith(".txt"):
	mata_data_file_path = os.path.join(subject_path, file)
	if video_file_path is None or mata_data_file_path is None:
	raise OSError("Files are incomplete in {}".format(subject_path))
	# Extract ground truth HR
	gtHR = None
	with open(mata_data_file_path, "r") as f:
	gtdata = [[float(l) for l in line.split()] for line in f.readlines()]
	gtHR = gtdata[1]
	if gtHR is None:
	raise ValueError("Ground truth heart rate value value is INCORRECT!")
	n_vid = len(gtHR) // 150
	for i in range(n_vid):
	start = i * 150
	end = (i + 1) * 150
	# Extract frames
	raw_frames = extract_video_frame(
	video_file_path, frame_range=(start, end), downsample=False
	)
	# Align face
	align_frames = align_face(raw_frames, fa)
	# Calculate gt
	cur_gtHR = np.mean(gtHR[start:end])
	# Save file
	des_path = os.path.join(des_path_root, subject)
	if not os.path.exists(des_path):
	os.makedirs(des_path)
	file_path = os.path.join(des_path, str(i))
	np.savez(file_path, frames=align_frames, hr=cur_gtHR)
	print(
	"{} saved with frames: {}; HR: {}!\n".format(
	file_path, align_frames.shape, cur_gtHR
	)
	)
	except (KeyboardInterrupt, SystemExit):
	raise
	except Exception as e:
	print(e, "\n")
	# Print origin trace info
	with open("./ubfc_rppg_dataset_error.txt", "a") as myfile:
	myfile.write("Session: {}\n".format(subject_path))
	exc_info = sys.exc_info()
	traceback.print_exception(*exc_info, file=myfile)
	myfile.write("\n")
	del exc_info
	duration = str(datetime.timedelta(seconds=time.time() - start))
	print("It takes {} time for extracting UBFC-rPPG dataset".format(duration))

Cheers