Building Floor detection

The detection of floors using conventional methods is difficult.

Requirements

python ==3.7.0 pytorch==1.6.0 opencv latest version torchvision==0.7.1 matplotlib latest version pillow

to train the model

python train.py --batch_size 3

to eval and visualize results

python eval.py

Name		Name	Last commit message	Last commit date
Latest commit History 18 Commits
checkpoints		checkpoints
dataset		dataset
log		log
plots		plots
predictions		predictions
README.md		README.md
auxilary.py		auxilary.py
calculate_means_stds.py		calculate_means_stds.py
dataset.py		dataset.py
eval.py		eval.py
loss.py		loss.py
model.py		model.py
train.py		train.py
train_other.py		train_other.py