search

ethz-asl / background_foreground_segmentation

12 stars 2 forks source link

Feature/ssseg dataloader #5

Closed renezurbruegg closed 3 years ago

renezurbruegg commented 3 years ago

Created a dataset loader to load images for self supervised training. This script assumes that training labels are stored int the workingDir path and validation labels from the hive are stored in the validationDir path.

renezurbruegg commented 3 years ago

run linter

renezurbruegg commented 3 years ago

run build

renezurbruegg commented 3 years ago

:+1: Better save than sorry. Also my first implementation would have matched wrong labels to images if there was a file missing which could have been disastrous.

I will rename the filenames, so if i plot the images/labels the file names will show if there is a missmatch

hermannsblum commented 3 years ago

@renezurbruegg can we merge this?

renezurbruegg commented 3 years ago

@renezurbruegg can we merge this?

I just committed a last change i did to train the model using the loader. If you agree with my changes we can merge it.

(Pycharm somehow messed up the formatting altough yapf is defined to run on save. If you disable whitespaces you can see the changes)