What's Changed

• Wall detection export rvt by @PawelKinczyk in #10
• Change files structure by @PawelKinczyk in #11

Full Changelog: v0.3.0...v0.4.0

Contributors

• @PawelKinczyk

What's Changed

• Add change RGB to grayscale cells by @PawelKinczyk in #7
• Add grayscale models by @PawelKinczyk in #8
• Two new models! Train on the same dataset but in grayscale. Below comparison on test data:

12 classes model

Main conclusion is that using gray images to train and detection increased the measures of small groups like "stairs", "refrigerator", "dishwasher" and "table".

Precision-Confidence Curve
a) RGB
image

b) Gray
image

Confusion matrix
a) RGB
image

b) Gray
image

F1 curve
a) RGB
image

b) Gray
image

Walls

Better results at 0.1 confidence. Interesting peak near 1.0 confidence in model_walls_gray.

Precision-Confidence Curve
a) RGB
image

b) Gray
image

Full Changelog: v0.2.0...v0.3.0

Contributors

• @PawelKinczyk

What's Changed

• Bump scipy from 1.9.1 to 1.10.0 by @dependabot in #3
• Bump tornado from 6.3.1 to 6.3.2 by @dependabot in #4
• Bump requests from 2.28.2 to 2.31.0 by @dependabot in #5
• Create new model by @PawelKinczyk in #6

New Contributors

• @dependabot made their first contribution in #3

Full Changelog: v0.1.0...v0.2.0

New wall detection model

I decided to train new model with less classes. I didn't change dataset since last release so i focus on biggest class. As i mention in readme about first model: "Walls are overrepresented because this is normal quantities in architectural plans."
image
This should improve of walls detection and YES! i get better results for walls with shorter train time (but train process could be longer because model is getting better even after 194 epoch). This is plot with "Train loss" :
image
And "Validation loss":
image

Validation

Confusion matrix:
image
Precision-Confidence curve:
image

Contributors

• @dependabot
• @PawelKinczyk

This is the first version of AECVision project. Available one model which detects 12 classes:

• 'bath',
• 'dishwasher / washing machine',
• 'door',
• 'furniture',
• 'refrigerator',
• 'sink',
• 'stairs',
• 'stove',
• 'table',
• 'wall',
• 'wc',
• 'window'