tusen-ai / simpledet

A Simple and Versatile Framework for Object Detection and Instance Recognition
Apache License 2.0
3.08k stars 486 forks source link

Segmentation Fault 11 when using mx.ndarray.argmax_channel #320

Open guoqiang0148666 opened 4 years ago

guoqiang0148666 commented 4 years ago

Thank your excellent works.

l train cascade_rcnn for two-class objection detection, after some epochs, l meet Segmentation Fault 11 when calculating 'RcnnAcc_1st', l debug the code and think the error may exist in the bellow code

In _core/detectionmetric.py

pred_label = mx.ndarray.argmax_channel(pred).astype('int32').asnumpy().reshape(-1)

l just wonder if it right and how to solver it.

Any suggestions will be welcomed.

Thanks