search

sunnyxiaohu / R-C3D.pytorch

R-C3D pytorch implementation
248 stars 67 forks source link

fix output_flat out of range write #30

Closed nihui closed 5 years ago

nihui commented 5 years ago

the output shape is [num_rois, num_channels, pooled_length, pooled_height, pooled_width]

fix segmentation fault when running on cpu