Skip to content

Commit 9bde2cc

Browse files
committed
imagenet-example-2: imageNet::Create does not accept enums as arguments
No version of imageNet::Create accepts enum (NetworkType) arguments. But is does accept a char*. Thus, replacing imageNet::GOOGLENET with "googlenet" does the trick. Signed-off-by: Shreeyash Pandey <[email protected]>
1 parent 667bb68 commit 9bde2cc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/imagenet-example-2.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,8 @@ int main( int argc, char** argv )
3939
}
4040

4141
// load the GoogleNet image recognition network with TensorRT
42-
// you can use imageNet::RESNET_18 to load ResNet-18 model instead
43-
imageNet* net = imageNet::Create(imageNet::GOOGLENET);
42+
// you can use "resnet-18" to load ResNet-18 model instead
43+
imageNet* net = imageNet::Create("googlenet");
4444

4545
// check to make sure that the network model loaded properly
4646
if( !net )

0 commit comments

Comments
 (0)