Fixed stream names in MPPImageClassifier
This commit is contained in:
parent
ddce041725
commit
1c4be91a3a
|
@ -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];
|
||||||
|
|
Loading…
Reference in New Issue
Block a user