Question

我编写了一个javascript程序，它使用遗传算法仅使用三角形重建图像。这是策略：

生成随机模型池，每个模型都有一个三角形阵列（3个点和一个颜色）
评估每个模型的适合度。为此，我将原始图像的像素阵列与我的模型进行比较。我使用Cosine Similarity来比较数组
保留最佳模型，并配对它们以创建新模型
随机改变一些模型
评估新池并继续

在一些迭代之后它可以很好地工作，你可以在这里看到：

我遇到的问题是，它非常慢，大部分时间花在获取模型的像素（将三角形列表（颜色+点）转换为像素数组）。我现在就是这样做的：

我的像素数组是一维数组，我需要能够将x，y坐标转换为索引：

static getIndex(x, y, width) {
  return 4 * (width * y + x);
}

然后我能够得出一点：

static plot(x, y, color, img) {
  let idx = this.getIndex(x, y, img.width);

  let added = [color.r, color.g, color.b, map(color.a, 0, 255, 0, 1)];
  let base = [img.pixels[idx], img.pixels[idx + 1], img.pixels[idx + 2], map(img.pixels[idx + 3], 0, 255, 0, 1)];
  let a01 = 1 - (1 - added[3]) * (1 - base[3]);

  img.pixels[idx + 0] = Math.round((added[0] * added[3] / a01) + (base[0] * base[3] * (1 - added[3]) / a01)); // red
  img.pixels[idx + 1] = Math.round((added[1] * added[3] / a01) + (base[1] * base[3] * (1 - added[3]) / a01)); // green
  img.pixels[idx + 2] = Math.round((added[2] * added[3] / a01) + (base[2] * base[3] * (1 - added[3]) / a01)); // blue
  img.pixels[idx + 3] = Math.round(map(a01, 0, 1, 0, 255));
}

然后一行：

 static line(x0, y0, x1, y1, img, color) {
  x0 = Math.round(x0);
  y0 = Math.round(y0);
  x1 = Math.round(x1);
  y1 = Math.round(y1);
  let dx = Math.abs(x1 - x0);
  let dy = Math.abs(y1 - y0);
  let sx = x0 < x1 ? 1 : -1;
  let sy = y0 < y1 ? 1 : -1;
  let err = dx - dy;

  do {
    this.plot(x0, y0, color, img);
    let e2 = 2 * err;
    if (e2 > -dy) {
      err -= dy;
      x0 += sx;
    }
    if (e2 < dx) {
      err += dx;
      y0 += sy;
    }
  } while (x0 != x1 || y0 != y1);
}

最后，一个三角形：

static drawTriangle(triangle, img) {
  for (let i = 0; i < triangle.points.length; i++) {
    let point = triangle.points[i];
    let p1 =
      i === triangle.points.length - 1
        ? triangle.points[0]
        : triangle.points[i + 1];
    this.line(point.x, point.y, p1.x, p1.y, img, triangle.color);
  }
  this.fillTriangle(triangle, img);
}

static fillTriangle(triangle, img) {
  let vertices = Array.from(triangle.points);
  vertices.sort((a, b) => a.y > b.y);
  if (vertices[1].y == vertices[2].y) {
    this.fillBottomFlatTriangle(vertices[0], vertices[1], vertices[2], img, triangle.color);
  } else if (vertices[0].y == vertices[1].y) {
    this.fillTopFlatTriangle(vertices[0], vertices[1], vertices[2], img, triangle.color);
  } else {
    let v4 = {
      x: vertices[0].x + float(vertices[1].y - vertices[0].y) / float(vertices[2].y - vertices[0].y) * (vertices[2].x - vertices[0].x),
    y: vertices[1].y
    };
    this.fillBottomFlatTriangle(vertices[0], vertices[1], v4, img, triangle.color);
    this.fillTopFlatTriangle(vertices[1], v4, vertices[2], img, triangle.color);
  }
}

static fillBottomFlatTriangle(v1, v2, v3, img, color) {
  let invslope1 = (v2.x - v1.x) / (v2.y - v1.y);
  let invslope2 = (v3.x - v1.x) / (v3.y - v1.y);

  let curx1 = v1.x;
  let curx2 = v1.x;

  for (let scanlineY = v1.y; scanlineY <= v2.y; scanlineY++) {
    this.line(curx1, scanlineY, curx2, scanlineY, img, color);
    curx1 += invslope1;
    curx2 += invslope2;
  }
}

static fillTopFlatTriangle(v1, v2, v3, img, color) {
  let invslope1 = (v3.x - v1.x) / (v3.y - v1.y);
  let invslope2 = (v3.x - v2.x) / (v3.y - v2.y);

  let curx1 = v3.x;
  let curx2 = v3.x;

  for (let scanlineY = v3.y; scanlineY > v1.y; scanlineY--) {
    this.line(curx1, scanlineY, curx2, scanlineY, img, color);
    curx1 -= invslope1;
    curx2 -= invslope2;
  }
}

您可以在行动here

中看到完整的代码

所以，我想知道：

是否可以优化此代码？
如果是，最好的方法是什么？也许有一个图书馆比我更好地完成所有绘图工作？或者通过使用工人？

谢谢！

Answer 1

我已经测试了你的建议，结果如下：

使用RMS而不是余弦相似性：如果相似性的度量更好，我不是一个人，但它确实不会更糟。它似乎运行得更快。
使用UInt8Array ：肯定会产生影响，但运行速度不会太快。虽然不慢。
绘制隐形画布：明确更快更轻松！我可以删除所有绘图函数，并用几行代码替换它，运行速度更快！

这是绘制到隐形画布的代码：

var canvas = document.createElement('canvas');
canvas.id = "CursorLayer";
canvas.width = this.width;
canvas.height = this.height;
canvas.display = "none";
var body = document.getElementsByTagName("body")[0];
body.appendChild(canvas);

var ctx = canvas.getContext("2d");
ctx.fillStyle = "rgba(0, 0, 0, 1)";
ctx.fillRect(0, 0, this.width, this.height);

for (let i = 0; i < this.items.length; i++) {
  let item = this.items[i];
  ctx.fillStyle = "rgba(" +item.color.r + ','+item.color.g+','+item.color.b+','+map(item.color.a, 0, 255, 0, 1)+")";
  ctx.beginPath();
  ctx.moveTo(item.points[0].x, item.points[0].y);
  ctx.lineTo(item.points[1].x, item.points[1].y);
  ctx.lineTo(item.points[2].x, item.points[2].y);
  ctx.fill();
}
let pixels = ctx.getImageData(0, 0, this.width, this.height).data;
//delete canvas
body.removeChild(canvas);
return pixels;

在这些更改之前，我的代码以每秒约1.68次迭代运行。现在它以每秒约16.45次迭代运行！

请参阅完整代码here。

再次感谢！

优化JavaScript的策略

1 个答案: