About running NMS with CPU

wizyoung / YOLOv3_TensorFlow

Complete YOLO v3 TensorFlow implementation. Support training on your own dataset.

MIT License

1.55k stars 579 forks source link

About running NMS with CPU #226

Open GlenZou opened 4 years ago

GlenZou commented 4 years ago

Thank you for your help！ After I changed gpu_nms in ‘test_single_image.py’ to cpu_nms, I got the error 'tensor' object has no attribute 'reshape'. Maybe I should convert the pred_box to ndarray, but I add .eval () and other errors will appear. what should I do?

qawsedrg commented 3 years ago

i have no idea which your code is. but i have succeeded to run nms on cpu, you may refer to https://github.com/qawsedrg/yolov3_yolov4_tensorflow, the full cpu_nms code is in the use_pb file without suffix "gpu"