import coremltools as ct
from coremltools.converters.mil.mil import types
coreml_model = ct.convert(
traced_model,
convert_to="mlprogram",
inputs=[ct.TensorType(shape=(1, 3, 16, 224, 224),dtype=types.float)]
)
coreml_model.save('yowo_tiny.mlmodel')
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\builder.py", line 168, in _add_op
new_op = op_cls(**kwargs)
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\operation.py", line 190, in init
self._validate_and_set_inputs(input_kv)
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\operation.py", line 503, in _validate_and_set_inputs
self.input_spec.validate_inputs(self.name, self.op_type, input_kvs)
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\input_type.py", line 163, in validate_inputs
raise ValueError(msg.format(name, var.name, input_type.type_str,
ValueError: Op "pred_reg.1" (op_type: gather) Input indices="anchor_idxs.1" expects tensor or scalar of dtype from type domain ['int32'] but got tensor[is1,fp32]
import coremltools as ct from coremltools.converters.mil.mil import types coreml_model = ct.convert( traced_model, convert_to="mlprogram", inputs=[ct.TensorType(shape=(1, 3, 16, 224, 224),dtype=types.float)] ) coreml_model.save('yowo_tiny.mlmodel')
File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\builder.py", line 168, in _add_op new_op = op_cls(**kwargs) File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\operation.py", line 190, in init self._validate_and_set_inputs(input_kv) File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\operation.py", line 503, in _validate_and_set_inputs self.input_spec.validate_inputs(self.name, self.op_type, input_kvs) File "C:\Users\Admin\AppData\Roaming\Python\Python310\site-packages\coremltools\converters\mil\mil\input_type.py", line 163, in validate_inputs raise ValueError(msg.format(name, var.name, input_type.type_str, ValueError: Op "pred_reg.1" (op_type: gather) Input indices="anchor_idxs.1" expects tensor or scalar of dtype from type domain ['int32'] but got tensor[is1,fp32]