Skip to content

Commit bc38f28

Browse files
committed
update default cnn classifier in config, small fixes.
1 parent aa3c915 commit bc38f28

File tree

3 files changed

+5
-6
lines changed

3 files changed

+5
-6
lines changed

camera.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ def __init__(self):
8989
if cnn_model != "":
9090
self._cnn_classifiers[cnn_model] = CNNManager.get_instance().load_model(cnn_model)
9191
self._cnn_classifier_default = self._cnn_classifiers[cnn_model]
92+
logging.info("loaded: " + cnn_model + " " + str(self._cnn_classifier_default))
9293

9394
self._camera.grab_start()
9495
self._image_cv = self.get_image()
@@ -365,11 +366,10 @@ def cnn_classify(self, model_name=None):
365366

366367
img = self.get_image()
367368
classes = classifier.classify_image(img.mat())
368-
s_classes = sorted(classes.items(), key=lambda x: x[1], reverse=True)
369-
return s_classes
369+
return classes
370370

371371
def find_class(self):
372-
return self.cnn_classify()[0][0]
372+
return self.cnn_classify()[0]
373373

374374
def sleep(self, elapse):
375375
logging.debug("sleep: " + str(elapse))

cnn_classifier.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,5 @@ def classify_image(self,
139139

140140
results = np.squeeze(results)
141141
result = results.argmax()
142-
pairs = {self._labels[result]: results[result]}
143-
logging.info(pairs)
142+
pairs = (self._labels[result], results[result])
144143
return pairs

coderbot.cfg

+1-1
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
{"move_power_angle_3": "60", "cnn_default_model": "fruit_025_128_1", "prog_maxblocks": "-1", "camera_jpeg_quality": "5", "show_page_control": "true", "camera_framerate": "30", "prog_scrollbars": "true", "move_fw_speed": "100", "prog_level": "adv", "move_motor_trim": "1", "move_motor_mode": "dc", "cv_image_factor": "2", "move_power_angle_1": "45", "camera_path_object_size_min": "4000", "button_func": "none", "camera_color_object_size_min": "4000", "camera_jpeg_bitrate": "1000000", "move_fw_elapse": "1", "show_control_move_commands": "true", "camera_color_object_size_max": "160000", "show_page_prefs": "true", "camera_exposure_mode": "auto", "ctrl_tr_elapse": "-1", "show_page_program": "true", "move_tr_elapse": "0.5", "camera_path_object_size_max": "160000", "sound_shutter": "$shutter.mp3", "ctrl_fw_elapse": "-1", "sound_stop": "$shutdown.mp3", "ctrl_tr_speed": "80", "ctrl_fw_speed": "100", "move_tr_speed": "85", "move_power_angle_2": "60", "ctrl_hud_image": "", "load_at_start": "", "sound_start": "$startup.mp3"}
1+
{"move_power_angle_3": "60", "cnn_default_model": "generic_fast_low", "prog_maxblocks": "-1", "camera_jpeg_quality": "5", "show_page_control": "true", "camera_framerate": "30", "prog_scrollbars": "true", "move_fw_speed": "100", "prog_level": "adv", "move_motor_trim": "1", "move_motor_mode": "dc", "cv_image_factor": "2", "move_power_angle_1": "45", "camera_path_object_size_min": "4000", "button_func": "none", "camera_color_object_size_min": "4000", "camera_jpeg_bitrate": "1000000", "move_fw_elapse": "1", "show_control_move_commands": "true", "camera_color_object_size_max": "160000", "show_page_prefs": "true", "camera_exposure_mode": "auto", "ctrl_tr_elapse": "-1", "show_page_program": "true", "move_tr_elapse": "0.5", "camera_path_object_size_max": "160000", "sound_shutter": "$shutter.mp3", "ctrl_fw_elapse": "-1", "sound_stop": "$shutdown.mp3", "ctrl_tr_speed": "80", "ctrl_fw_speed": "100", "move_tr_speed": "85", "move_power_angle_2": "60", "ctrl_hud_image": "", "load_at_start": "", "sound_start": "$startup.mp3"}

0 commit comments

Comments
 (0)