search

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
https://github.com/hunglc007/tensorflow-yolov4-tflite
MIT License
2.23k stars 1.24k forks source link

convert_trt.py - FileNotFoundError for 'coco_dataset/coco/5k.txt' #373

Closed kaijaz123 closed 3 years ago

kaijaz123 commented 3 years ago 
Script:
!python convert_trt.py --weights ./checkpoints/yolov4-tiny-ball.tf --quantize_mode int8 --output ./checkpoints/tiny-ball-416-trt-int8

Error:
FileNotFoundError: [Errno 2] No such file or directory: '/media/user/Source/Data/coco_dataset/coco/5k.txt'

When I try with quantize_mode float16 it works but crash on int8.

has anyone encountered this problem before while converting trt mode?

Thanks.

abhi-84 commented 2 years ago

I am also facing same issue. have u resolved it?

kaijaz123 commented 2 years ago

Unfortunately no. As the error stated, we need to find a way to get the txt file for the coco dataset.