Fixed bug where models would evaluate without specifying any flags (#71)

This commit is contained in:
Ryan Lai 2018-10-26 10:50:39 -07:00 коммит произвёл GitHub
Родитель d7492d3da8
Коммит 47ed2f5bba
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 4 добавлений и 5 удалений

Просмотреть файл

@ -419,14 +419,13 @@ int main(int argc, char** argv)
{
output.SetDefaultCSVFileName();
}
std::vector<DeviceType> deviceTypes = FetchDeviceTypes(args);
std::vector<InputBindingType> inputBindingTypes = FetchInputBindingTypes(args);
std::vector<InputDataType> inputDataTypes = FetchInputDataTypes(args);
std::vector<std::wstring> modelPaths = args.ModelPath().empty() ? GetModelsInDirectory(args, &output) : std::vector<std::wstring>(1, args.ModelPath());
if (!args.ModelPath().empty() || !args.FolderPath().empty())
{
std::vector<DeviceType> deviceTypes = FetchDeviceTypes(args);
std::vector<InputBindingType> inputBindingTypes = FetchInputBindingTypes(args);
std::vector<InputDataType> inputDataTypes = FetchInputDataTypes(args);
std::vector<std::wstring> modelPaths = args.ModelPath().empty() ? GetModelsInDirectory(args, &output) : std::vector<std::wstring>(1, args.ModelPath());
return EvaluateModels(modelPaths, deviceTypes, inputBindingTypes, inputDataTypes, args, output);
}