Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import mxnet as mx
- pretrained_net = mx.gluon.model_zoo.vision.get_model(name='inceptionv3', pretrained=True, classes=1000, prefix='aligcroc_')
- net = mx.gluon.model_zoo.vision.get_model(name='inceptionv3', classes=2, prefix='aligcroc_')
- net.features = pretrained_net.features
- net.output.initialize()
- batch_size = 1
- channels = 3
- height = width = 299
- data_batch = mx.ndarray.random.normal(shape=(batch_size, channels, height, width))
- net(data_batch)
Add Comment
Please, Sign In to add comment