byzhang

Results 2 issues of byzhang

当输入宽高不一致时报错, 在训练过程中报错,报错时机随缘: yolo_head.py", line 351, in get_assignments bboxes_preds_per_image = bboxes_preds_per_image[fg_mask] RuntimeError: assertion `dtype == dst.dtype && dst.is_contiguous()' failed at ../../../../../../dnn/src/common/elemwise/opr_impl.cpp:281: void megdnn::ElemwiseForward::check_layout_and_broadcast(const TensorLayoutPtrArray&, const megdnn::TensorLayout&)

输入图片尺寸为1704*1280时出现错误: malloc_consolidate(): invalid chunk size Aborted (core dumped) 将 std::memcpy(mask.data, mask_v.data(), mask_v.size()* sizeof(int)); 改为 std::memcpy(mask.data, mask_v.data(), (img_h * img_w) * sizeof(int)); 以mask的元素数量拷贝,解决此问题

contributor