Skip to content

Commit 8125ec5

Browse files
authored
TFLite --int8 'flatbuffers==1.12' fix (ultralytics#6216)
* TFLite `--int8` 'flatbuffers==1.12' fix Temporary workaround for TFLite INT8 export. * Update export.py * Update export.py
1 parent 5402753 commit 8125ec5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

export.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -277,8 +277,6 @@ def export_tflite(keras_model, im, file, int8, data, ncalib, prefix=colorstr('Te
277277
try:
278278
import tensorflow as tf
279279

280-
from models.tf import representative_dataset_gen
281-
282280
LOGGER.info(f'\n{prefix} starting export with tensorflow {tf.__version__}...')
283281
batch_size, ch, *imgsz = list(im.shape) # BCHW
284282
f = str(file).replace('.pt', '-fp16.tflite')
@@ -288,6 +286,8 @@ def export_tflite(keras_model, im, file, int8, data, ncalib, prefix=colorstr('Te
288286
converter.target_spec.supported_types = [tf.float16]
289287
converter.optimizations = [tf.lite.Optimize.DEFAULT]
290288
if int8:
289+
from models.tf import representative_dataset_gen
290+
check_requirements(('flatbuffers==1.12',)) # https://github.com/ultralytics/yolov5/issues/5707
291291
dataset = LoadImages(check_dataset(data)['train'], img_size=imgsz, auto=False) # representative data
292292
converter.representative_dataset = lambda: representative_dataset_gen(dataset, ncalib)
293293
converter.target_spec.supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS_INT8]

0 commit comments

Comments
 (0)