bc - Image classifier example now includes OccupancyClassifier

NHERI-SimCenter · Apr 11, 2024 · ed1442f · ed1442f
1 parent 5c5c86e
commit ed1442f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/examples/image_processor/brails_image_classifier.py b/examples/image_processor/brails_image_classifier.py
@@ -51,8 +51,7 @@
 aerial_images.set_directory("./images/satellite", True)
 street_images.set_directory("./images/street", True)
 
-# Test importer and one of the aerial imagery classifiers:
-
+# Test importer and one of the aerial imagery classifiers, RoofShapeClassifier:
 my_class = importer.get_class('RoofShapeClassifier')
 my_classifier = my_class()
 predictions = my_classifier.predict(aerial_images)
@@ -63,8 +62,9 @@
 predictions = my_classifier.predict(aerial_images)
 print(predictions)
 
-my_class = importer.get_class('FacadeParser')
+# Test importer and one of the street-level imagery classifiers, 
+# OccupancyClassifier:
+my_class = importer.get_class('OccupancyClassifier')
 my_classifier = my_class()
-# will need a new image set
 predictions = my_classifier.predict(street_images) 
-print(predictions)
+print(predictions)