working

Author: vpandiarajan20

cli/ml_inference.go

@@ -34,14 +34,6 @@ func MLInferenceInferAction(c *cli.Context, args mlInferenceInferArgs) error {
 		return err
 	}
 
-	// Print the arguments
-	fmt.Println("OrgID: ", args.OrgID)
-	fmt.Println("FileOrgID: ", args.FileOrgID)
-	fmt.Println("FileID: ", args.FileID)
-	fmt.Println("FileLocationID: ", args.FileLocationID)
-	fmt.Println("ModelID: ", args.ModelID)
-	fmt.Println("ModelVersion: ", args.ModelVersion)
-
 	_, err = client.mlRunInference(
 		args.OrgID, args.FileOrgID, args.FileID, args.FileLocationID,
 		args.ModelID, args.ModelVersion)
@@ -72,7 +64,51 @@ func (c *viamClient) mlRunInference(orgID, fileOrgID, fileID, fileLocation, mode
 	if err != nil {
 		return nil, errors.Wrapf(err, "received error from server")
 	}
-	fmt.Println("OutputTensors: ", resp.OutputTensors)
-	fmt.Println("Annotations: ", resp.Annotations)
+	printInferenceResponse(resp)
 	return resp, nil
 }
+
+// printInferenceResponse prints a neat representation of the GetInferenceResponse.
+func printInferenceResponse(resp *mlinferencepb.GetInferenceResponse) {
+	fmt.Println("Inference Response:")
+	fmt.Println("Output Tensors:")
+	if resp.OutputTensors != nil {
+		for name, tensor := range resp.OutputTensors.Tensors {
+			fmt.Printf("  Tensor Name: %s\n", name)
+			fmt.Printf("    Shape: %v\n", tensor.Shape)
+			if tensor.Tensor != nil {
+				fmt.Print("    Values: [")
+				for i, value := range tensor.GetDoubleTensor().GetData() {
+					if i > 0 {
+						fmt.Print(", ")
+					}
+					fmt.Printf("%.4f", value)
+				}
+				fmt.Println("]")
+			} else {
+				fmt.Println("    No values available.")
+			}
+		}
+	} else {
+		fmt.Println("  No output tensors.")
+	}
+
+	fmt.Println("Annotations:")
+	if resp.Annotations != nil {
+		for _, bbox := range resp.Annotations.Bboxes {
+			fmt.Printf("  Bounding Box ID: %s, Label: %s\n", bbox.Id, bbox.Label)
+			fmt.Printf("    Coordinates: [%f, %f, %f, %f]\n", bbox.XMinNormalized, bbox.YMinNormalized, bbox.XMaxNormalized, bbox.YMaxNormalized)
+			if bbox.Confidence != nil {
+				fmt.Printf("    Confidence: %.4f\n", *bbox.Confidence)
+			}
+		}
+		for _, classification := range resp.Annotations.Classifications {
+			fmt.Printf("  Classification Label: %s\n", classification.Label)
+			if classification.Confidence != nil {
+				fmt.Printf("    Confidence: %.4f\n", *classification.Confidence)
+			}
+		}
+	} else {
+		fmt.Println("  No annotations.")
+	}
+}