Update Src/Classifier.cpp

Src/Classifier.cpp

@@ -41,14 +41,11 @@ ErrorCode Classifier::Initialize(InitializationParameterOfClassifier initializat
     OrtROCMProviderOptions rocm_options;
     rocm_options.device_id = 0;
     sessionOptions.AppendExecutionProvider_ROCM(rocm_options);
-    
     sessionOptions.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_BASIC);
     // sessionOptions.EnableProfiling("profile_prefix");
 
     session = new Ort::Session(env, modelPath.c_str(), sessionOptions);
-
     return SUCCESS;
-
 }
 
 ErrorCode Classifier::Classify(const std::vector<cv::Mat> &srcImages,std::vector<std::vector<ResultOfPrediction>> &predictions)
@@ -117,23 +114,19 @@ ErrorCode Classifier::Classify(const std::vector<cv::Mat> &srcImages,std::vector
     for(int i=0;i<srcImages.size();++i)
     {
         int startIndex=numberOfClasses*i;
-
         std::vector<float> logit;
         for(int j=0;j<numberOfClasses;++j)
         {
             logit.push_back(pdata[startIndex+j]);
-        }
-        
+        }     
         std::vector<ResultOfPrediction> resultOfPredictions;
         for(int j=0;j<numberOfClasses;++j)
         {
             ResultOfPrediction prediction;
             prediction.label=j;
             prediction.confidence=logit[j];
-
             resultOfPredictions.push_back(prediction);
         }
-
         predictions.push_back(resultOfPredictions);
     }
     return SUCCESS;