message: Added separate classes for category mask and confidence mask for more clarity.
diff: NS_ASSUME_NONNULL_BEGIN
  +/** Holds a confidence mask belonging to a single class and its meta data. */
  +@interface TFLConfidenceMask : NSObject
  +
  +/**
  + * Confidence masks of size `width` x `height` for any one class.
  + */
  +@property(nonatomic, assign) float *mask;
  +
  +/**
  + * The width of the mask. This is an intrinsic parameter of the model being
  + * used, and does not depend on the input image dimensions.
  + */
  +@property(nonatomic, assign) NSInteger width;
  +
  +/**
  + * The height of the mask. This is an intrinsic parameter of the model being
  + * used, and does not depend on the input image dimensions.
  + */
  +@property(nonatomic, assign) NSInteger height;
  +
  +@end
  +
  +/** Holds category mask and its metadata. */
  +@interface TFLCategoryMask : NSObject
  +
  +/**
  + * Flattened 2D-array of size `width` x `height`, in row major order.
  + * The value of each pixel in this mask represents the class to which the
  + * pixel belongs.
  + */
  +@property(nonatomic, assign) UInt8 *mask;
  +
  +/**
  + * The width of the mask. This is an intrinsic parameter of the model being
  + * used, and does not depend on the input image dimensions.
  + */
  +@property(nonatomic, assign) NSInteger width;
  +
  +/**
  + * The height of the mask. This is an intrinsic parameter of the model being
  + * used, and does not depend on the input image dimensions.
  + */
  +@property(nonatomic, assign) NSInteger height;
  +
  +@end
  +
  /** Holds a label associated with an RGB color, for display purposes. */
  @interface TFLColoredLabel : NSObject
  @@ -41,30 +87,20 @@ NS_ASSUME_NONNULL_BEGIN
  @interface TFLSegmentation : NSObject
  /**
  - * The width of the mask. This is an intrinsic parameter of the model being
  - * used, and does not depend on the input image dimensions.
  - */
  -@property(nonatomic, assign) NSInteger width;
  -
  -/**
  - * The height of the mask. This is an intrinsic parameter of the model being
  - * used, and does not depend on the input image dimensions.
  - */
  -@property(nonatomic, assign) NSInteger height;
  -
  -/**
  - * One confidence masks of size `width` x `height` for each of the supported
  - * classes. The value of each pixel in these masks represents the confidence
  - * score for this particular class.
  + * Array of confidence masks where each element is a confidence mask of size
  + * `width` x `height`, one for each of the supported classes.
  + * The value of each pixel in these masks represents the confidence score for
  + * this particular class.
  + * This property is mutually exclusive with `categoryMask`.
  */
  -@property(nonatomic, assign) float *_Nullable *_Nullable confidenceMasks;
  +@property(nonatomic, copy, nullable) NSArray<TFLConfidenceMask *> *confidenceMasks;
  -/** Flattened 2D-array of size `width` x `height`, in row major order.
  +/** Holds the category mask .
  * The value of each pixel in this mask represents the class to which the
  * pixel belongs.
  + * This property is mutually exclusive with `confidenceMasks`.
  */
  -@property(nonatomic, assign, nullable) NSUInteger *categoryMask;
  -
  +@property(nonatomic, copy, nullable) TFLCategoryMask *categoryMask;
  /**
  * The list of colored labels for all the supported categories (classes).
  