diff --git a/export.py b/export.py index 55908c075f95..a825a73b2d3c 100644 --- a/export.py +++ b/export.py @@ -180,7 +180,7 @@ def export_openvino(model, file, half, prefix=colorstr('OpenVINO:')): cmd = f"mo --input_model {file.with_suffix('.onnx')} --output_dir {f} --data_type {'FP16' if half else 'FP32'}" subprocess.check_output(cmd.split()) # export - with open(Path(f) / file.with_suffix('.yaml'), 'w') as g: + with open(Path(f) / file.with_suffix('.yaml').name, 'w') as g: yaml.dump({'stride': int(max(model.stride)), 'names': model.names}, g) # add metadata.yaml LOGGER.info(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)') diff --git a/models/common.py b/models/common.py index 547a4c946a9f..bad55b4024c0 100644 --- a/models/common.py +++ b/models/common.py @@ -368,7 +368,7 @@ def __init__(self, weights='yolov5s.pt', device=torch.device('cpu'), dnn=False, network = ie.read_model(model=w, weights=Path(w).with_suffix('.bin')) executable_network = ie.compile_model(model=network, device_name="CPU") output_layer = next(iter(executable_network.outputs)) - meta = w.with_suffix('.yaml') + meta = Path(w).with_suffix('.yaml') if meta.exists(): stride, names = self._load_metadata(meta) # load metadata elif engine: # TensorRT