|
106 | 106 | </span></span><span style=display:flex><span> gray <span style=color:#f92672>=</span> cv2<span style=color:#f92672>.</span>cvtColor(img, cv2<span style=color:#f92672>.</span>COLOR_BGR2GRAY) |
107 | 107 | </span></span><span style=display:flex><span> faces <span style=color:#f92672>=</span> face_cascade<span style=color:#f92672>.</span>detectMultiScale(gray, <span style=color:#ae81ff>1.1</span>, <span style=color:#ae81ff>4</span>) |
108 | 108 | </span></span><span style=display:flex><span> <span style=color:#66d9ef>for</span> (x, y, w, h) <span style=color:#f92672>in</span> faces: |
109 | | -</span></span><span style=display:flex><span> cv2<span style=color:#f92672>.</span>rectangle(img, (x, y), (x<span style=color:#f92672>+</span>w, y<span style=color:#f92672>+</span>h), (<span style=color:#ae81ff>255</span>, <span style=color:#ae81ff>0</span>, <span style=color:#ae81ff>0</span>), <span style=color:#ae81ff>2</span>) |
| 109 | +</span></span><span style=display:flex><span> cv2<span style=color:#f92672>.</span>rectangle(img, (x, y), (x <span style=color:#f92672>+</span> w, y <span style=color:#f92672>+</span> h), (<span style=color:#ae81ff>255</span>, <span style=color:#ae81ff>0</span>, <span style=color:#ae81ff>0</span>), <span style=color:#ae81ff>2</span>) |
110 | 110 | </span></span><span style=display:flex><span> <span style=color:#75715e># Display the output</span> |
111 | 111 | </span></span><span style=display:flex><span> cv2<span style=color:#f92672>.</span>imshow(<span style=color:#e6db74>'img'</span>, img) |
112 | 112 | </span></span><span style=display:flex><span> |
113 | 113 | </span></span><span style=display:flex><span><span style=color:#75715e># Update image in interface</span> |
114 | 114 | </span></span><span style=display:flex><span><span style=color:#66d9ef>def</span> <span style=color:#a6e22e>updateImage</span>(image): |
115 | | -</span></span><span style=display:flex><span> _interfaces[<span style=color:#ae81ff>0</span>]<span style=color:#f92672>.</span>setImage(OpenCVUtils<span style=color:#f92672>.</span>convertImageToML(image), minMaxValues <span style=color:#f92672>=</span> [<span style=color:#ae81ff>0</span>,<span style=color:#ae81ff>255</span>]) |
| 115 | +</span></span><span style=display:flex><span> _interfaces[<span style=color:#ae81ff>0</span>]<span style=color:#f92672>.</span>setImage(OpenCVUtils<span style=color:#f92672>.</span>convertImageToML(image), minMaxValues <span style=color:#f92672>=</span> [<span style=color:#ae81ff>0</span>, <span style=color:#ae81ff>255</span>]) |
116 | 116 | </span></span><span style=display:flex><span> |
117 | 117 | </span></span><span style=display:flex><span><span style=color:#75715e># Start capturing webcam</span> |
118 | 118 | </span></span><span style=display:flex><span><span style=color:#66d9ef>def</span> <span style=color:#a6e22e>startCapture</span>(): |
|
167 | 167 | </span></span><span style=display:flex><span> <span style=color:#75715e># Step 1: Get image from webcam capture</span> |
168 | 168 | </span></span><span style=display:flex><span> stopCapture() |
169 | 169 | </span></span><span style=display:flex><span> inImage <span style=color:#f92672>=</span> ctx<span style=color:#f92672>.</span>field(<span style=color:#e6db74>"PythonImage.output0"</span>)<span style=color:#f92672>.</span>image() |
170 | | -</span></span><span style=display:flex><span> img <span style=color:#f92672>=</span> inImage<span style=color:#f92672>.</span>getTile((<span style=color:#ae81ff>0</span>,<span style=color:#ae81ff>0</span>,<span style=color:#ae81ff>0</span>,<span style=color:#ae81ff>0</span>,<span style=color:#ae81ff>0</span>,<span style=color:#ae81ff>0</span>), inImage<span style=color:#f92672>.</span>imageExtent())[<span style=color:#ae81ff>0</span>,<span style=color:#ae81ff>0</span>,:,<span style=color:#ae81ff>0</span>,:,:] |
| 170 | +</span></span><span style=display:flex><span> img <span style=color:#f92672>=</span> inImage<span style=color:#f92672>.</span>getTile((<span style=color:#ae81ff>0</span>, <span style=color:#ae81ff>0</span>, <span style=color:#ae81ff>0</span>, <span style=color:#ae81ff>0</span>, <span style=color:#ae81ff>0</span>, <span style=color:#ae81ff>0</span>), inImage<span style=color:#f92672>.</span>imageExtent())[<span style=color:#ae81ff>0</span>, <span style=color:#ae81ff>0</span>, :, <span style=color:#ae81ff>0</span>, :, :] |
171 | 171 | </span></span><span style=display:flex><span> |
172 | 172 | </span></span><span style=display:flex><span> <span style=color:#75715e># Step 2: Convert image into torch tensor</span> |
173 | 173 | </span></span><span style=display:flex><span> img <span style=color:#f92672>=</span> torch<span style=color:#f92672>.</span>Tensor(img)<span style=color:#f92672>.</span>type(torch<span style=color:#f92672>.</span>uint8) |
|
0 commit comments