BBuf / cv_tools

62 stars 17 forks source link

batch!=1 #2

Open shzql opened 3 years ago

shzql commented 3 years ago

对于batch不等于1的应该怎么修改

for (int i = 0; i < output_shape.size(); i++) {
        auto masks = g_masks[i];
        vector<vector<int> > anchors;
        for (auto mask : masks)
            anchors.push_back(g_anchors[mask]);
        auto shape = shapes[i];
        for (int h = 0; h < shape[0]; h++) {
            int offset_h = offset + h * shape[1] * shape[2] * shape[3];
            for (int w = 0; w < shape[1]; w++) {
                int offset_w = offset_h + w * shape[2] * shape[3];
                for (int c = 0; c < shape[2]; c++) {

上面没看到跟batch相关的呀?