cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed stream names in MPPImageClassifier

Browse Source
This commit is contained in:
Prianka Liz Kariat 2023-03-23 18:41:27 +05:30
parent ddce041725
commit 1c4be91a3a
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mediapipe/tasks/ios/vision/image_classifier/sources/MPPImageClassifier.mm
View File

@ -36,7 +36,7 @@ static NSString *const kClassificationsTag = @"CLASSIFICATIONS";
static NSString *const kImageInStreamName = @"image_in"; static NSString *const kImageInStreamName = @"image_in";
static NSString *const kImageOutStreamName = @"image_out"; static NSString *const kImageOutStreamName = @"image_out";
static NSString *const kImageTag = @"IMAGE"; static NSString *const kImageTag = @"IMAGE";
static NSString *const kNormRectName = @"norm_rect_in"; static NSString *const kNormRectStreamName = @"norm_rect_in";
static NSString *const kNormRectTag = @"NORM_RECT"; static NSString *const kNormRectTag = @"NORM_RECT";
static NSString *const kTaskGraphName = static NSString *const kTaskGraphName =
@ -44,7 +44,7 @@ static NSString *const kTaskGraphName =
#define InputPacketMap(imagePacket, normalizedRectPacket) \ #define InputPacketMap(imagePacket, normalizedRectPacket) \
{ \ { \
{kImageInStreamName.cppString, imagePacket}, { kNormRectName.cppString, normalizedRectPacket } \ {kImageInStreamName.cppString, imagePacket}, { kNormRectStreamName.cppString, normalizedRectPacket } \
} }
@interface MPPImageClassifier () { @interface MPPImageClassifier () {
@ -61,9 +61,13 @@ static NSString *const kTaskGraphName =
MPPTaskInfo *taskInfo = [[MPPTaskInfo alloc] MPPTaskInfo *taskInfo = [[MPPTaskInfo alloc]
initWithTaskGraphName:kTaskGraphName initWithTaskGraphName:kTaskGraphName
inputStreams:@[ [NSString inputStreams:@[ [NSString
stringWithFormat:@"%@:%@", kImageTag, kImageInStreamName] ] stringWithFormat:@"%@:%@", kImageTag, kImageInStreamName],
[NSString
stringWithFormat:@"%@:%@", kNormRectTag, kNormRectStreamName] ]
outputStreams:@[ [NSString stringWithFormat:@"%@:%@", kClassificationsTag, outputStreams:@[ [NSString stringWithFormat:@"%@:%@", kClassificationsTag,
kClassificationsStreamName] ] kClassificationsStreamName],
[NSString stringWithFormat:@"%@:%@", kImageTag,
kImageOutStreamName] ]
taskOptions:options taskOptions:options
enableFlowLimiting:NO enableFlowLimiting:NO
error:error]; error:error];