mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
More cleanup
* Moved variable inits around. * Added spacing.
This commit is contained in:
parent
19473ef28c
commit
107019a357
@ -27,10 +27,10 @@ STATIC mp_obj_t py_net_forward(uint n_args, const mp_obj_t *args, mp_map_t *kw_a
|
|||||||
{
|
{
|
||||||
nn_t *net = py_net_cobj(args[0]);
|
nn_t *net = py_net_cobj(args[0]);
|
||||||
image_t *img = py_helper_arg_to_image_mutable(args[1]);
|
image_t *img = py_helper_arg_to_image_mutable(args[1]);
|
||||||
|
bool softmax = py_helper_keyword_int(n_args, args, 2, kw_args, MP_OBJ_NEW_QSTR(MP_QSTR_softmax), false);
|
||||||
|
bool dry_run = py_helper_keyword_int(n_args, args, 3, kw_args, MP_OBJ_NEW_QSTR(MP_QSTR_dry_run), false);
|
||||||
|
|
||||||
mp_obj_t output_list = mp_obj_new_list(0, NULL);
|
mp_obj_t output_list = mp_obj_new_list(0, NULL);
|
||||||
bool softmax = py_helper_keyword_int(n_args, args, 2, kw_args, MP_OBJ_NEW_QSTR(MP_QSTR_softmax), false);
|
|
||||||
bool dry_run = py_helper_keyword_int(n_args, args, 3, kw_args, MP_OBJ_NEW_QSTR(MP_QSTR_dry_run), false);
|
|
||||||
|
|
||||||
if (dry_run == false) {
|
if (dry_run == false) {
|
||||||
nn_run_network(net, img, softmax);
|
nn_run_network(net, img, softmax);
|
||||||
@ -41,6 +41,7 @@ STATIC mp_obj_t py_net_forward(uint n_args, const mp_obj_t *args, mp_map_t *kw_a
|
|||||||
for (int i=0; i<net->output_size; i++) {
|
for (int i=0; i<net->output_size; i++) {
|
||||||
mp_obj_list_append(output_list, mp_obj_new_int(net->output_data[i]));
|
mp_obj_list_append(output_list, mp_obj_new_int(net->output_data[i]));
|
||||||
}
|
}
|
||||||
|
|
||||||
return output_list;
|
return output_list;
|
||||||
}
|
}
|
||||||
STATIC MP_DEFINE_CONST_FUN_OBJ_KW(py_net_forward_obj, 2, py_net_forward);
|
STATIC MP_DEFINE_CONST_FUN_OBJ_KW(py_net_forward_obj, 2, py_net_forward);
|
||||||
@ -66,9 +67,8 @@ static const mp_obj_type_t py_net_type = {
|
|||||||
|
|
||||||
static mp_obj_t py_nn_load(mp_obj_t path_obj)
|
static mp_obj_t py_nn_load(mp_obj_t path_obj)
|
||||||
{
|
{
|
||||||
py_net_obj_t *net = NULL;
|
|
||||||
const char *path = mp_obj_str_get_str(path_obj);
|
const char *path = mp_obj_str_get_str(path_obj);
|
||||||
net = m_new_obj(py_net_obj_t);
|
py_net_obj_t *net = m_new_obj(py_net_obj_t);
|
||||||
net->base.type = &py_net_type;
|
net->base.type = &py_net_type;
|
||||||
nn_load_network(py_net_cobj(net), path);
|
nn_load_network(py_net_cobj(net), path);
|
||||||
return net;
|
return net;
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user