diff --git a/cnn_model.c b/cnn_model.c index 40e10c1..5bf3e42 100644 --- a/cnn_model.c +++ b/cnn_model.c @@ -164,12 +164,13 @@ u8 model_write(char* model_name) u32 _larr = 0; u8 _len = strlen(model_name); char _path[_len+1+7+30]; + char _datapath[_len+1+7+30]; char _fstr[READLENGTH+1] = {0}; int progress; Model *_model = model(model_name); if(_model == NULL || strcmp(model_name, "data") == 0){ - sprintf(_path, "./dataset/%s.txt", model_name); + sprintf(_path, "./dataset/_data/%s.txt", model_name); FILE *file = fopen(_path, "r"); if(file == NULL){ DEBUG_PRINTF("\r\n输入了一个无效的模型或Data数据集的名字\r\n"); @@ -194,7 +195,8 @@ u8 model_write(char* model_name) return 200; } - sprintf(_path, "./dataset/%s.txt", _model -> dname ? _model -> dname : _model -> name); + if(_model -> dname)sprintf(_datapath, "_data/%s", _model -> dname); + sprintf(_path, "./dataset/%s.txt", _model -> dname ? _datapath : _model -> name); FILE *file = fopen(_path, "r"); if(file == NULL){ DEBUG_PRINTF("文件[%s]无法打开\r\n", _model -> dname ? _model -> dname : _model -> name); @@ -270,7 +272,7 @@ u8 model_switchdata(char* data_name){ u8 _len = strlen(data_name); char _path[_len+1+7]; if(data.array != NULL)modelmym_free("data"); - sprintf(_path, "./dataset/%s.txt", data_name); + sprintf(_path, "./dataset/_data/%s.txt", data_name); FILE *file = fopen(_path, "r"); if(file == NULL){ DEBUG_PRINTF("\r\nData数据集[%s]不存在\r\n",data_name);