def CreateNet(net, overwrite=False, input_blobs=None):
if input_blobs is None:
input_blobs = []
for input_blob in input_blobs:
C.create_blob(input_blob)
return C.create_net(StringifyProto(net), overwrite)
def CreateNet(net, input_blobs=None):
if input_blobs is None:
input_blobs = []
for input_blob in input_blobs:
C.create_blob(input_blob)
return C.create_net(StringfyProto(net))
C.set_global_engine_pref({caffe2_pb2.HIP : engine_list})
print('Using MIOPEN')
C.feed_blob('data', img, device_opts.SerializeToString())
init_def = caffe2_pb2.NetDef()
with open(INIT_NET, 'rb') as f:
init_def.ParseFromString(f.read())
init_def.device_option.CopyFrom(device_opts)
C.run_net_once(init_def.SerializeToString())
net_def = caffe2_pb2.NetDef()
with open(PREDICT_NET, 'rb') as f:
net_def.ParseFromString(f.read())
net_def.device_option.CopyFrom(device_opts)
C.create_net(net_def.SerializeToString())
C.feed_blob('data', img, device_opts.SerializeToString())
### Debug code
"""
for b in workspace.Blobs():
x = workspace.FetchBlob(b)
if type(x) != str:
print(str(b) + ': ' + str(x.shape))
"""
### End of debug code
print('Running net ' + workspace.GetNetName(net_def) + '...')
C.run_net(workspace.GetNetName(net_def), 1, False)