Skip to content
Advertisement

Tag: keras

How to fix failed assertion `output channels should be divisible by group’ when trying to fit the model in Keras?

I’m trying to use ImageDataGenerator() for my image datasets. Here is my image augmentation code: Then use that plug into my model: Use EarlyStopping: Compile and Fit the model: That is when the code crash, and gives this error message. I try to change the output neurons but that doesn’t work. I don’t know what to do anymore. Please help

Convert tfrecords to image

I found a training dataset which is a set of tfrecords files,im trying to convert them into images but with no results,is it possible to convert them to images ? Answer To find out what is inside a tf.record use tf.data.TFRecordDataset and tf.train.Example: To parse the records, use tf.data.TFRecordDataset with tf.io.parse_single_example and tf.io.parse_tensor: Also check the source code of Satellite

Import onnx models to tensorflow2.x?

I created a modified lenet model using tensorflow that looks like this: When I finish training I save the model using tf.keras.models.save_model : Then I transform this model into onnx format using “tf2onnx” module: I want a method that can retrieve the same model into tensorflow2.x. I tried to use “onnx_tf” to transform the onnx model into tensorflow .pb model:

Advertisement