Skip to content

The demo.py file has some issues related to the use of textual_prompts. #46

@xzh929

Description

@xzh929

The textual_prompts variable in demo.py is a 1D array, but the set_ensemble_text_prompts function in model.py fails to process it correctly; it requires a 2D array.
The result before modifying the code is shown in the figure below:

used ensemble text prompts ===========
det prompts: c
filtered background: o
det prompts: c
filtered background: a
======================================
candle,, 4, 1, 0.3, 0.25, 0.0

Image

The result after changing the code is shown in the figure below:

used ensemble text prompts ===========
det prompts: color defect. hole. black defect. wick hole. spot. 
filtered background: candle
======================================
candle,, 4, 1, 0.3, 0.25, 0.0

Image

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions