allanzelener / YAD2K

YAD2K: Yet Another Darknet 2 Keras
Other
2.71k stars 879 forks source link

Difference between yad2k.py and yad2k/models/keras_yolo.py? #176

Open goodwillyoga opened 4 years ago

goodwillyoga commented 4 years ago

Greetings!

Hopefully, someone would be able to help me understand the difference between yad2k.py vs keras_yolo.py.

On the onset, it seems like yad2k.py reads Darknet19 config and weights and creates Keras model with TF backend whereas keras_yolo.py creates the YOLO_v2 Model Defined in Keras.

In keras_yolo.py the library is using partial, func_tools to extend Kera's Conv2D and define convenient methods that return a compose layers equivalent to Darkent config sections like DarknetConv2D, DarknetConv2D_BN_Leaky, darknet_body.

Anything that adds more to understand will be very helpful. It's interesting that both files were initially checked in on Feb 14th, 2017 (must be a lucky day :-))

Thanks