Skip to content

eluv-io/model-celeb

Repository files navigation

Setup

With Podman

Dependencies

  1. Podman with nvidia toolkit enabled
  2. Python
  3. Access to eluv-io repo via ssh key

Download celebrity detection model

python download_weights.py

Add ssh keys to ssh-agent

ssh-add (on personal machine)

NOTE: if you are on a remote server, either you should have your ssh key on the remote server and run ssh-add there, or you should run it on your personal machine and verify that you are connected with agent forwarding enabled.

Build image

./build.sh

Default run

podman run --rm --volume=$(pwd)/test:/elv/test:ro --volume=$(pwd)/tags:/elv/tags --volume=$(pwd)/.cache:/root/.cache --network host --device nvidia.com/gpu=0 celeb test/1.mp4 test/2.mp4
  1. Note: you must mount the files to tag into the container storage (--volume=$(pwd)/test:/elv/test)
  2. Tag files will appear in the tags directory (--volume=$(pwd)/tags:/elv/tags).

Custom run

  1. Default parameters are found in config.yml under runtime/default
  2. These values can be overriden by passing in data with --config when running the container in the command line:
podman run --rm --volume=$(pwd)/test:/elv/test:ro --volume=$(pwd)/tags:/elv/tags --volume=$(pwd)/.cache:/root/.cache --network host --device nvidia.com/gpu=0 celeb test/1.mp4 test/2.mp4 --config '{"fps":8, "content_id":"<qid>", "thres":0.6}'

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Packages

 
 
 

Contributors