index.js 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. var tf = require("@tensorflow/tfjs-core");
  4. var classes_1 = require("../classes");
  5. var Dimensions_1 = require("../classes/Dimensions");
  6. function isTensor(tensor, dim) {
  7. return tensor instanceof tf.Tensor && tensor.shape.length === dim;
  8. }
  9. exports.isTensor = isTensor;
  10. function isTensor1D(tensor) {
  11. return isTensor(tensor, 1);
  12. }
  13. exports.isTensor1D = isTensor1D;
  14. function isTensor2D(tensor) {
  15. return isTensor(tensor, 2);
  16. }
  17. exports.isTensor2D = isTensor2D;
  18. function isTensor3D(tensor) {
  19. return isTensor(tensor, 3);
  20. }
  21. exports.isTensor3D = isTensor3D;
  22. function isTensor4D(tensor) {
  23. return isTensor(tensor, 4);
  24. }
  25. exports.isTensor4D = isTensor4D;
  26. function isFloat(num) {
  27. return num % 1 !== 0;
  28. }
  29. exports.isFloat = isFloat;
  30. function isEven(num) {
  31. return num % 2 === 0;
  32. }
  33. exports.isEven = isEven;
  34. function round(num, prec) {
  35. if (prec === void 0) { prec = 2; }
  36. var f = Math.pow(10, prec);
  37. return Math.floor(num * f) / f;
  38. }
  39. exports.round = round;
  40. function isDimensions(obj) {
  41. return obj && obj.width && obj.height;
  42. }
  43. exports.isDimensions = isDimensions;
  44. function computeReshapedDimensions(_a, inputSize) {
  45. var width = _a.width, height = _a.height;
  46. var scale = inputSize / Math.max(height, width);
  47. return new Dimensions_1.Dimensions(Math.round(width * scale), Math.round(height * scale));
  48. }
  49. exports.computeReshapedDimensions = computeReshapedDimensions;
  50. function getCenterPoint(pts) {
  51. return pts.reduce(function (sum, pt) { return sum.add(pt); }, new classes_1.Point(0, 0))
  52. .div(new classes_1.Point(pts.length, pts.length));
  53. }
  54. exports.getCenterPoint = getCenterPoint;
  55. function range(num, start, step) {
  56. return Array(num).fill(0).map(function (_, i) { return start + (i * step); });
  57. }
  58. exports.range = range;
  59. function isValidNumber(num) {
  60. return !!num && num !== Infinity && num !== -Infinity && !isNaN(num) || num === 0;
  61. }
  62. exports.isValidNumber = isValidNumber;
  63. function isValidProbablitiy(num) {
  64. return isValidNumber(num) && 0 <= num && num <= 1.0;
  65. }
  66. exports.isValidProbablitiy = isValidProbablitiy;
  67. //# sourceMappingURL=index.js.map