11package mmdeploy ;
22
3+ /** @description: the Java API class of Classifier. */
34public class Classifier {
45 static {
56 System .loadLibrary ("mmdeploy_java" );
67 }
78
89 private final long handle ;
910
11+ /** @description: Single classification result of a picture. */
1012 public static class Result {
13+
14+ /** Class id. */
1115 public int label_id ;
16+
17+ /** Class score. */
1218 public float score ;
19+
20+ /** Initializes a new instance of the Result class.
21+ * @param label_id: class id.
22+ * @param score: class score.
23+ */
1324 public Result (int label_id , float score ) {
1425 this .label_id = label_id ;
1526 this .score = score ;
1627 }
1728 }
1829
19- public Classifier (String modelPath , String deviceName , int deviceId ) {
30+ /** Initializes a new instance of the Classifier class.
31+ * @param modelPath: model path.
32+ * @param deviceName: device name.
33+ * @param deviceId: device ID.
34+ * @exception Exception: create Classifier failed exception.
35+ */
36+ public Classifier (String modelPath , String deviceName , int deviceId ) throws Exception {
2037 handle = create (modelPath , deviceName , deviceId );
38+ if (handle == -1 ) {
39+ throw new Exception ("Create Classifier failed!" );
40+ }
2141 }
2242
23- public Result [][] apply (Mat [] images ) {
43+ /** Get label information of each image in a batch.
44+ * @param images: input mats.
45+ * @return: results of each input mat.
46+ * @exception Exception: apply Classifier failed exception.
47+ */
48+ public Result [][] apply (Mat [] images ) throws Exception {
2449 int [] counts = new int [images .length ];
2550 Result [] results = apply (handle , images , counts );
51+ if (results == null ) {
52+ throw new Exception ("Apply Classifier failed!" );
53+ }
2654 Result [][] rets = new Result [images .length ][];
2755 int offset = 0 ;
2856 for (int i = 0 ; i < images .length ; ++i ) {
@@ -36,12 +64,22 @@ public Result[][] apply(Mat[] images) {
3664 return rets ;
3765 }
3866
39- public Result [] apply (Mat image ) {
67+ /** Get label information of one image.
68+ * @param image: input mat.
69+ * @return: result of input mat.
70+ * @exception Exception: apply Classifier failed exception.
71+ */
72+ public Result [] apply (Mat image ) throws Exception {
4073 int [] counts = new int [1 ];
4174 Mat [] images = new Mat []{image };
42- return apply (handle , images , counts );
75+ Result [] results = apply (handle , images , counts );
76+ if (results == null ) {
77+ throw new Exception ("Apply Classifier failed!" );
78+ }
79+ return results ;
4380 }
4481
82+ /** Release the instance of Classifier. */
4583 public void release () {
4684 destroy (handle );
4785 }
0 commit comments