본문

yolov5 export.py error 변환 오류 에러


yolov5 export.py error 변환 오류 에러_1.png

TensorFlow SavedModel: export failure 1.4s: name 'TFMaxPool2d' is not defined


말그대로 TFXXXX 뭐시기가 구현안된거니 

yolov5/models/tf.py에 해당 구현하면 끝


MaxPool2d 일경우


class TFMaxPool2d(keras.layers.Layer):
    def __init__(self, k=(2, 2), s=None, p='same', w=None):
        super().__init__()
        if p == 0:
            padding = 'valid'
        else:
            padding = 'same'
        self.m = keras.layers.MaxPool2D(pool_size=k, strides=s, padding=padding)

    def call(self, inputs):
        return self.m(inputs)

추가하면됨 


출처:https://github.com/ultralytics/yolov5/issues/7091