Fixed wrong cast for XNOR-net

build/darknet/x64/partial.cmd

@@ -9,6 +9,9 @@ rem darknet.exe partial cfg/tiny-yolo-voc.cfg tiny-yolo-voc.weights tiny-yolo-vo
 darknet.exe partial cfg/darknet53_448.cfg darknet53_448.weights darknet53.conv.74 74
 
 
+darknet.exe partial cfg/darknet53_448_xnor.cfg darknet53_448_xnor.weights darknet53_448_xnor.conv.74 74
+
+
 darknet.exe partial cfg/yolov2-tiny-voc.cfg yolov2-tiny-voc.weights yolov2-tiny-voc.conv.13 13
 
 

src/gemm.c

@@ -1646,7 +1646,7 @@ void im2col_cpu_custom_bin(float* data_im,
                     __m256 result256 = _mm256_cmp_ps(src256, float_zero256, _CMP_GT_OS);
                     uint16_t mask = _mm256_movemask_ps(result256); // (val > 0) ? 0 : 1
 
-                    uint16_t* dst_ptr = &((uint16_t*)data_col)[col_index / 8];
+                    uint16_t* dst_ptr = (uint16_t*)&((uint8_t*)data_col)[col_index / 8];
                     *dst_ptr |= (mask << (col_index % 8));
                 }