search

akashgokul / VP4LaneDet

Vanishing Point for Lane Detection
17 stars 4 forks source link

self.num_test_samples and self.num_train_samples #1

Closed dyh closed 4 years ago

dyh commented 4 years ago

https://github.com/akashgokul/294_FinalProj/blob/47b0c56c24e754e28312f0fcc57cb03126386a0e/torch/dataset.py#L54-L56

maybe switch them: [self.num_train_samples:] <--> [:self.num_train_samples]

self.train_img_names = train_and_valid_img_names.iloc[:self.num_train_samples] self.validation_img_names = train_and_valid_img_names.iloc[self.num_train_samples:]

akashgokul commented 4 years ago

Yes, you are correct.

akashgokul commented 3 years ago

The colab notebooks, have the most up to date code without this bug. Sorry!