Skip to content

coremltools-0.6.3

Compare
Choose a tag to compare
@srikris srikris released this 01 Sep 04:17

Features

  • Linux support
  • Added a “useCPUOnly” flag that lets you run predictions using CoreML through Python bindings using only the CPU

Note: coremltools-0.6.2 has a known issue with the useCPUOnly flag that failed on certain neural network models. This has been fixed with 0.6.3

Neural Network Builder

Added support for layers in the NeuralNetworkBuilder that were present in the neural network protobuf but missing from the builder:

  • Local response normalization (LRN) layer
  • Split layer
  • Unary function layer
  • Bias, scale layers
  • Load constant layer
  • L2 normalization layer
  • Mean variance normalization (MVN) layer
  • Elementwise min layer
  • Depthwise and separable convolutions

Added support for some of the missing parameters in NeuralNetworkBuilder:

  • Padding options in convolution, pooling and padding layers
  • Scale and shift options for linear activation

Other bug fixes & enhancements

  • Bug-fix in the caffe converter that was preventing the elementwise max layer from converting.
  • Support for converting DepthwiseConv2D and SeparableConv2D from Keras