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

how to change default max=200 to arbitrary value in the last yolo layer in conv_lbbox #339

Open yingtina opened 3 years ago

yingtina commented 3 years ago

If the objects to detect in the image are more than 200, I need to add max=1000 e.g. in the last yolo layer in .cfg from AlexeyAB. How to add this line in tensorflow framework? I checked the def convolutional in common.py, but max is not a paramter for convolutional definition. Could you please give me some hint? Thanks.

yingtina commented 3 years ago

@hunglc007 Could you please give me some hint?