webgl_morphtargets_webcam.html 9.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <title>three.js webgl - morph targets - webcam</title>
  5. <meta charset="utf-8">
  6. <meta name="viewport" content="width=device-width, user-scalable=no, minimum-scale=1.0, maximum-scale=1.0">
  7. <link type="text/css" rel="stylesheet" href="main.css">
  8. <style>
  9. body {
  10. background-color: #666666;
  11. }
  12. </style>
  13. </head>
  14. <body>
  15. <div id="info">
  16. <a href="https://threejs.org" target="_blank" rel="noopener">three.js</a> webgl - morph targets - webcam<br/>
  17. model by <a href="https://www.bannaflak.com/face-cap" target="_blank" rel="noopener">Face Cap</a>
  18. </div>
  19. <script type="importmap">
  20. {
  21. "imports": {
  22. "three": "../build/three.module.js",
  23. "three/addons/": "./jsm/"
  24. }
  25. }
  26. </script>
  27. <script type="module">
  28. import * as THREE from 'three';
  29. import { OrbitControls } from 'three/addons/controls/OrbitControls.js';
  30. import { RoomEnvironment } from 'three/addons/environments/RoomEnvironment.js';
  31. import { GLTFLoader } from 'three/addons/loaders/GLTFLoader.js';
  32. import { KTX2Loader } from 'three/addons/loaders/KTX2Loader.js';
  33. import { MeshoptDecoder } from 'three/addons/libs/meshopt_decoder.module.js';
  34. import { GUI } from 'three/addons/libs/lil-gui.module.min.js';
  35. // Mediapipe
  36. import vision from 'https://cdn.jsdelivr.net/npm/@mediapipe/tasks-vision@0.10.0';
  37. const { FaceLandmarker, FilesetResolver } = vision;
  38. const blendshapesMap = {
  39. // '_neutral': '',
  40. 'browDownLeft': 'browDown_L',
  41. 'browDownRight': 'browDown_R',
  42. 'browInnerUp': 'browInnerUp',
  43. 'browOuterUpLeft': 'browOuterUp_L',
  44. 'browOuterUpRight': 'browOuterUp_R',
  45. 'cheekPuff': 'cheekPuff',
  46. 'cheekSquintLeft': 'cheekSquint_L',
  47. 'cheekSquintRight': 'cheekSquint_R',
  48. 'eyeBlinkLeft': 'eyeBlink_L',
  49. 'eyeBlinkRight': 'eyeBlink_R',
  50. 'eyeLookDownLeft': 'eyeLookDown_L',
  51. 'eyeLookDownRight': 'eyeLookDown_R',
  52. 'eyeLookInLeft': 'eyeLookIn_L',
  53. 'eyeLookInRight': 'eyeLookIn_R',
  54. 'eyeLookOutLeft': 'eyeLookOut_L',
  55. 'eyeLookOutRight': 'eyeLookOut_R',
  56. 'eyeLookUpLeft': 'eyeLookUp_L',
  57. 'eyeLookUpRight': 'eyeLookUp_R',
  58. 'eyeSquintLeft': 'eyeSquint_L',
  59. 'eyeSquintRight': 'eyeSquint_R',
  60. 'eyeWideLeft': 'eyeWide_L',
  61. 'eyeWideRight': 'eyeWide_R',
  62. 'jawForward': 'jawForward',
  63. 'jawLeft': 'jawLeft',
  64. 'jawOpen': 'jawOpen',
  65. 'jawRight': 'jawRight',
  66. 'mouthClose': 'mouthClose',
  67. 'mouthDimpleLeft': 'mouthDimple_L',
  68. 'mouthDimpleRight': 'mouthDimple_R',
  69. 'mouthFrownLeft': 'mouthFrown_L',
  70. 'mouthFrownRight': 'mouthFrown_R',
  71. 'mouthFunnel': 'mouthFunnel',
  72. 'mouthLeft': 'mouthLeft',
  73. 'mouthLowerDownLeft': 'mouthLowerDown_L',
  74. 'mouthLowerDownRight': 'mouthLowerDown_R',
  75. 'mouthPressLeft': 'mouthPress_L',
  76. 'mouthPressRight': 'mouthPress_R',
  77. 'mouthPucker': 'mouthPucker',
  78. 'mouthRight': 'mouthRight',
  79. 'mouthRollLower': 'mouthRollLower',
  80. 'mouthRollUpper': 'mouthRollUpper',
  81. 'mouthShrugLower': 'mouthShrugLower',
  82. 'mouthShrugUpper': 'mouthShrugUpper',
  83. 'mouthSmileLeft': 'mouthSmile_L',
  84. 'mouthSmileRight': 'mouthSmile_R',
  85. 'mouthStretchLeft': 'mouthStretch_L',
  86. 'mouthStretchRight': 'mouthStretch_R',
  87. 'mouthUpperUpLeft': 'mouthUpperUp_L',
  88. 'mouthUpperUpRight': 'mouthUpperUp_R',
  89. 'noseSneerLeft': 'noseSneer_L',
  90. 'noseSneerRight': 'noseSneer_R',
  91. // '': 'tongueOut'
  92. };
  93. //
  94. const renderer = new THREE.WebGLRenderer( { antialias: true } );
  95. renderer.setPixelRatio( window.devicePixelRatio );
  96. renderer.setSize( window.innerWidth, window.innerHeight );
  97. renderer.toneMapping = THREE.ACESFilmicToneMapping;
  98. document.body.appendChild( renderer.domElement );
  99. const camera = new THREE.PerspectiveCamera( 60, window.innerWidth / window.innerHeight, 1, 100 );
  100. camera.position.z = 5;
  101. const scene = new THREE.Scene();
  102. scene.scale.x = - 1;
  103. const environment = new RoomEnvironment();
  104. const pmremGenerator = new THREE.PMREMGenerator( renderer );
  105. scene.background = new THREE.Color( 0x666666 );
  106. scene.environment = pmremGenerator.fromScene( environment ).texture;
  107. const controls = new OrbitControls( camera, renderer.domElement );
  108. // Face
  109. let face, eyeL, eyeR;
  110. const eyeRotationLimit = THREE.MathUtils.degToRad( 30 );
  111. const ktx2Loader = new KTX2Loader()
  112. .setTranscoderPath( 'jsm/libs/basis/' )
  113. .detectSupport( renderer );
  114. new GLTFLoader()
  115. .setKTX2Loader( ktx2Loader )
  116. .setMeshoptDecoder( MeshoptDecoder )
  117. .load( 'models/gltf/facecap.glb', ( gltf ) => {
  118. const mesh = gltf.scene.children[ 0 ];
  119. scene.add( mesh );
  120. const head = mesh.getObjectByName( 'mesh_2' );
  121. head.material = new THREE.MeshNormalMaterial();
  122. face = mesh.getObjectByName( 'mesh_2' );
  123. eyeL = mesh.getObjectByName( 'eyeLeft' );
  124. eyeR = mesh.getObjectByName( 'eyeRight' );
  125. // GUI
  126. const gui = new GUI();
  127. gui.close();
  128. const influences = head.morphTargetInfluences;
  129. for ( const [ key, value ] of Object.entries( head.morphTargetDictionary ) ) {
  130. gui.add( influences, value, 0, 1, 0.01 )
  131. .name( key.replace( 'blendShape1.', '' ) )
  132. .listen( influences );
  133. }
  134. renderer.setAnimationLoop( animate );
  135. } );
  136. // Video Texture
  137. const video = document.createElement( 'video' );
  138. const texture = new THREE.VideoTexture( video );
  139. texture.colorSpace = THREE.SRGBColorSpace;
  140. const geometry = new THREE.PlaneGeometry( 1, 1 );
  141. const material = new THREE.MeshBasicMaterial( { map: texture, depthWrite: false } );
  142. const videomesh = new THREE.Mesh( geometry, material );
  143. scene.add( videomesh );
  144. // MediaPipe
  145. const filesetResolver = await FilesetResolver.forVisionTasks(
  146. 'https://cdn.jsdelivr.net/npm/@mediapipe/tasks-vision@0.10.0/wasm'
  147. );
  148. const faceLandmarker = await FaceLandmarker.createFromOptions( filesetResolver, {
  149. baseOptions: {
  150. modelAssetPath: 'https://storage.googleapis.com/mediapipe-models/face_landmarker/face_landmarker/float16/1/face_landmarker.task',
  151. delegate: 'GPU'
  152. },
  153. outputFaceBlendshapes: true,
  154. outputFacialTransformationMatrixes: true,
  155. runningMode: 'VIDEO',
  156. numFaces: 1
  157. } );
  158. if ( navigator.mediaDevices && navigator.mediaDevices.getUserMedia ) {
  159. navigator.mediaDevices.getUserMedia( { video: { facingMode: 'user' } } )
  160. .then( function ( stream ) {
  161. video.srcObject = stream;
  162. video.play();
  163. } )
  164. .catch( function ( error ) {
  165. console.error( 'Unable to access the camera/webcam.', error );
  166. } );
  167. }
  168. const transform = new THREE.Object3D();
  169. function animate() {
  170. if ( video.readyState >= HTMLMediaElement.HAVE_METADATA ) {
  171. const results = faceLandmarker.detectForVideo( video, Date.now() );
  172. if ( results.facialTransformationMatrixes.length > 0 ) {
  173. const facialTransformationMatrixes = results.facialTransformationMatrixes[ 0 ].data;
  174. transform.matrix.fromArray( facialTransformationMatrixes );
  175. transform.matrix.decompose( transform.position, transform.quaternion, transform.scale );
  176. const object = scene.getObjectByName( 'grp_transform' );
  177. object.position.x = transform.position.x;
  178. object.position.y = transform.position.z + 40;
  179. object.position.z = - transform.position.y;
  180. object.rotation.x = transform.rotation.x;
  181. object.rotation.y = transform.rotation.z;
  182. object.rotation.z = - transform.rotation.y;
  183. }
  184. if ( results.faceBlendshapes.length > 0 ) {
  185. const faceBlendshapes = results.faceBlendshapes[ 0 ].categories;
  186. // Morph values does not exist on the eye meshes, so we map the eyes blendshape score into rotation values
  187. const eyeScore = {
  188. leftHorizontal: 0,
  189. rightHorizontal: 0,
  190. leftVertical: 0,
  191. rightVertical: 0,
  192. };
  193. for ( const blendshape of faceBlendshapes ) {
  194. const categoryName = blendshape.categoryName;
  195. const score = blendshape.score;
  196. const index = face.morphTargetDictionary[ blendshapesMap[ categoryName ] ];
  197. if ( index !== undefined ) {
  198. face.morphTargetInfluences[ index ] = score;
  199. }
  200. // There are two blendshape for movement on each axis (up/down , in/out)
  201. // Add one and subtract the other to get the final score in -1 to 1 range
  202. switch ( categoryName ) {
  203. case 'eyeLookInLeft':
  204. eyeScore.leftHorizontal += score;
  205. break;
  206. case 'eyeLookOutLeft':
  207. eyeScore.leftHorizontal -= score;
  208. break;
  209. case 'eyeLookInRight':
  210. eyeScore.rightHorizontal -= score;
  211. break;
  212. case 'eyeLookOutRight':
  213. eyeScore.rightHorizontal += score;
  214. break;
  215. case 'eyeLookUpLeft':
  216. eyeScore.leftVertical -= score;
  217. break;
  218. case 'eyeLookDownLeft':
  219. eyeScore.leftVertical += score;
  220. break;
  221. case 'eyeLookUpRight':
  222. eyeScore.rightVertical -= score;
  223. break;
  224. case 'eyeLookDownRight':
  225. eyeScore.rightVertical += score;
  226. break;
  227. }
  228. }
  229. eyeL.rotation.z = eyeScore.leftHorizontal * eyeRotationLimit;
  230. eyeR.rotation.z = eyeScore.rightHorizontal * eyeRotationLimit;
  231. eyeL.rotation.x = eyeScore.leftVertical * eyeRotationLimit;
  232. eyeR.rotation.x = eyeScore.rightVertical * eyeRotationLimit;
  233. }
  234. }
  235. videomesh.scale.x = video.videoWidth / 100;
  236. videomesh.scale.y = video.videoHeight / 100;
  237. renderer.render( scene, camera );
  238. controls.update();
  239. }
  240. window.addEventListener( 'resize', function () {
  241. camera.aspect = window.innerWidth / window.innerHeight;
  242. camera.updateProjectionMatrix();
  243. renderer.setSize( window.innerWidth, window.innerHeight );
  244. } );
  245. </script>
  246. </body>
  247. </html>