chengjunwen / spp_net

spp net, tensorflow 实现
24 stars 13 forks source link

fix two bugs #4

Closed xiaoland closed 9 months ago

xiaoland commented 4 years ago

修复两个错误: 1、当batch_size为None时,进行reshape会报错 2、进行maxpool时,ksize和strides的输入有问题,已经修改,并且换成了tf.keras.MaxPool2D的最大池化操作