allo- / virtual_webcam_background

Use a virtual webcam background and overlays with body-pix and v4l2loopback
GNU General Public License v3.0
306 stars 48 forks source link

Resnet50 uses stride 16 when stride 32 was specified #86

Closed raffaem closed 2 years ago

raffaem commented 2 years ago
(virtualwebcam) [raffaele@fedora virtual_webcam_background]$ ./get-model-full.sh bodypix/resnet50/float/model-stride32
mkdir: cannot create directory ‘bodypix_resnet50_float_model-stride32’: File exists
(virtualwebcam) [raffaele@fedora virtual_webcam_background]$ cat config.yaml
model: resnet50
output_stride: 32
segmentation_threshold: 0.75
blur: 3
erode: 10
dilate: 10
virtual_video_device: "/dev/video0"
real_video_device: "/dev/video1"
average_masks: 3
mjpeg: False
layers:
  - "empty": [["video", "/home/raffaele/Pictures/webcam_video.webm"]]
  - "foreground": []
(virtualwebcam) [raffaele@fedora virtual_webcam_background]$ python3 virtual_webcam.py 
Reloading config.
Model: resnet50 (stride=16)
Loading model...
done.
Loading video: /home/raffaele/Pictures/webcam_video.webm
raffaem commented 2 years ago

The "Advanced" section of README.md is wrong.

The parameter is called stride, not output_stride.

Anyway the 16 version is much better :)