nestauk / asf_floorplan_interpreter

Modelling to interpret floor plan images to extract or infer information about a property's layout.

MIT License

0 stars 0 forks source link

Finalise scoping #10

Closed lizgzil closed 11 months ago

lizgzil commented 11 months ago

Description

The output of these changes is 2 trained models:

models/window_door_config_yolov8m_wd/: This is the model trained on around 100 floorplans labelled using Prodigy, plus the RoboFlow dataset. Merging these datasets was done in the new script merge_prodigy_roboflow.py. Only WINDOW and DOOR classes are used and trained.
models/room_config_yolov8m/: This is the model trained on ROOMs annotated in around 100 floorplans labelled using Prodigy.

In the future:

Label more ROOMs
Edit prodigy_to_yolo.py to input the newest room dataset and convert it to yolo format.
Train the room model again

Fixes # (issue)

Instructions for Reviewer

In order to test the code in this PR you need to ...

Please pay special attention to ...

Checklist:

[ ] I have refactored my code out from notebooks/
[ ] I have checked the code runs
[ ] I have tested the code
[ ] I have run pre-commit and addressed any issues not automatically fixed
[ ] I have merged any new changes from dev
[ ] I have documented the code
- [ ] Major functions have docstrings
- [ ] Appropriate information has been added to READMEs
[ ] I have explained this PR above
[ ] I have requested a code review

lizgzil commented 11 months ago

thanks @cmbrennan002 ! I've addressed your comments