I couldn’t figure out what caused the error “boxes must be 2-D[54] [Op:NonMaxSuppressionV3]”
return scores, boxes, classes
InvalidArgumentError: boxes must be 2-D[54] [Op:NonMaxSuppressionV3]
[ 6.0797963 3.743308 1.3914018 -0.34089637]], shape=(54, 4), dtype=float32)