BrianG13 / FLEX

FLEX: Parameter-free Multi-view 3D Human Motion Reconstruction
47 stars 3 forks source link

Train frames for Clips #19

Closed anas-zafar closed 1 year ago

anas-zafar commented 1 year ago

Hi @BrianG13 , I am trying to train FLEX on my custom dataset. I am a bit confused about the code below:

offset = 10
train_frames = random.randint(10, 50) * 4 if self.config.trainer.train_frames == 0 else self.config.trainer.train_frames
train_frames = 196
self.sequence_index = np.array(slice_set(offset, train_frames, self.frame_numbers))
self.r_sequence_index = np.array(slice_set(offset, train_frames,  self.r_frame_numbers)) if self.is_train else 0

My dataset contains maximum 200 frames per video. Should I change my offset and train frames? Thanks

BrianG13 commented 1 year ago

You can leave it this way, should be OK