logger.info("Loading weights from: {}".format(weights_file))
ws_blobs = workspace.Blobs()
with open(weights_file, "r") as f:
src_blobs = pickle.load(f)
if "cfg" in src_blobs:
saved_cfg = load_cfg(src_blobs["cfg"])
configure_bbox_reg_weights(model, saved_cfg)
if "blobs" in src_blobs:
After Change
logger.info("Loading weights from: {}".format(weights_file))
ws_blobs = workspace.Blobs()
src_blobs = load_object(weights_file)
if "cfg" in src_blobs:
saved_cfg = load_cfg(src_blobs["cfg"])
configure_bbox_reg_weights(model, saved_cfg)