lamnguyenkhoa / container-code-recognition

Detect and extract containers code in a video.

Hi there 👋

container-code-recognition's Issues

Need better OCR

Module error when run

Hello lamnguyenkhoa,

Im getting error when run:

python single_frame_process.py -i images\container1.png -c yolov4.cfg -w yolov4.weights -cl yolov4.txt

Error:
code_region_detector/build_model(): loading YOLO from disk... Traceback (most recent call last): File "C:\Users\brgus\Downloads\container-code-recognition-main\single_frame_process.py", line 86, in <module> main(argument) File "C:\Users\brgus\Downloads\container-code-recognition-main\single_frame_process.py", line 73, in main net, classes, output_layers = build_model(args['classes'], args['weights'], args['config']) File "C:\Users\brgus\Downloads\container-code-recognition-main\src\code_region_detector.py", line 32, in build_model output_layers = [layer_names[i[0] - 1] for i in net.getUnconnectedOutLayers()] # Get the yolo layer File "C:\Users\brgus\Downloads\container-code-recognition-main\src\code_region_detector.py", line 32, in <listcomp> output_layers = [layer_names[i[0] - 1] for i in net.getUnconnectedOutLayers()] # Get the yolo layer IndexError: invalid index to scalar variable.

IndexError: invalid index to scalar variable.

I have error:
"IndexError: invalid index to scalar variable."

How to fix it. Thanks