nestauk / asf_floorplan_interpreter

Modelling to interpret floor plan images to extract or infer information about a property's layout.
MIT License
1 stars 1 forks source link

Evaluation and full pipeline #17

Closed lizgzil closed 11 months ago

lizgzil commented 11 months ago

Description

Quite a few changes in one PR - I'm sorry! A summary:

Fixes #15 #14 #13

Instructions for Reviewer

There are two main things

  1. Can you check evaluate.py is how you expect it to be and I've not made any errors
  2. Try to run:
from asf_floorplan_interpreter.pipeline.predict_floorplan import FloorplanPredictor
img = 'outputs/figures/floorplan.png' # Local directory or a URL to an image file
fp = FloorplanPredictor(labels_to_predict = ["WINDOW", "DOOR","KITCHEN", "LIVING", "RESTROOM", "BEDROOM", "GARAGE"])
fp.load(local=True)
labels, label_counts = fp.predict_labels(img, conf_threshold=0)
fp.plot(img, labels, "outputs/figures/floorplan_prediction.png", plot_label=False)

Checklist:

lizgzil commented 11 months ago

Thanks @cmbrennan002 ! the changes you suggested are in this commit https://github.com/nestauk/asf_floorplan_interpreter/pull/17/commits/4fc1e1c8477700ebbb154b0a78290b48ccd2de37

@sqr00t It might also be worth you checking over that commit to see whether any of the new changes are things you need to know about!