RobotEdh / Yolov-4

Yolo v4 using TensorFlow 2.x
110 stars 30 forks source link

Requirements #12

Open danielmimimi opened 3 years ago

danielmimimi commented 3 years ago

Hello,

nice implementation, unfortunately, I can't let it run due to module installations. I Installed following packages as suggested.

keras == 2.2.4 tensorflow-gpu == 2.1.0

and get_custom_objects().update({'mish': Mish(mish)}) throws the AttributeError: module 'tensorflow' has no attribute 'get_default_graph error.

is there a requirements.txt?

Any help appreciated.