pull/4269/head
AlexeyAB 6 years ago
parent f8c72acd42
commit 69b7a19f5b
  1. 3
      src/data.c
  2. 2
      src/parser.c

@ -841,7 +841,8 @@ data load_data_detection(int n, char **paths, int m, int w, int h, int c, int bo
d.y = make_matrix(n, 5*boxes);
int i_mixup = 0;
for (i_mixup = 0; i_mixup <= mixup; i_mixup++) {
if (i_mixup) augmentation_calculated = 0;
if (i_mixup) augmentation_calculated = 0; // recalculate augmentation for the 2nd sequence if(track==1)
for (i = 0; i < n; ++i) {
float *truth = (float*)calloc(5 * boxes, sizeof(float));
const char *filename = (i_mixup) ? mixup_random_paths[i] : random_paths[i];

@ -409,7 +409,7 @@ int *parse_gaussian_yolo_mask(char *a, int *num) // Gaussian_YOLOv3
for (i = 0; i < len; ++i) {
if (a[i] == ',') ++n;
}
mask = calloc(n, sizeof(int));
mask = (int *)calloc(n, sizeof(int));
for (i = 0; i < n; ++i) {
int val = atoi(a);
mask[i] = val;

Loading…
Cancel
Save