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); d.y = make_matrix(n, 5*boxes);
int i_mixup = 0; int i_mixup = 0;
for (i_mixup = 0; i_mixup <= mixup; i_mixup++) { 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) { for (i = 0; i < n; ++i) {
float *truth = (float*)calloc(5 * boxes, sizeof(float)); float *truth = (float*)calloc(5 * boxes, sizeof(float));
const char *filename = (i_mixup) ? mixup_random_paths[i] : random_paths[i]; 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) { for (i = 0; i < len; ++i) {
if (a[i] == ',') ++n; if (a[i] == ',') ++n;
} }
mask = calloc(n, sizeof(int)); mask = (int *)calloc(n, sizeof(int));
for (i = 0; i < n; ++i) { for (i = 0; i < n; ++i) {
int val = atoi(a); int val = atoi(a);
mask[i] = val; mask[i] = val;

Loading…
Cancel
Save