Skip to content

Conversation

@johndpope
Copy link

@johndpope johndpope commented Nov 15, 2024

pip install git+https://github.com/johndpope/L2CS-Net.git@main

# Will automatically download gaze360 model if not present
pipeline = Pipeline(
    model_type='gaze360',  # or 'mpiigaze'
    arch='ResNet50',
    device='cuda'
)

# Or use custom weights
pipeline = Pipeline(
    weights=Path('path/to/custom/weights.pkl'),
    arch='ResNet50',
    device='cuda'
)

- Move softmax and idx_tensor initialization outside of include_detector conditional
- Fixes AttributeError when using Pipeline without face detector
- Required for integration with VASA-1 which uses its own face detection
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant