Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feature/advanced configuration update #47

Merged
merged 12 commits into from
Mar 26, 2024
Merged

Conversation

HonzaCuhel
Copy link
Contributor

This PR includes the following new pipeline arguments:

  • --annotation_iou_threshold: Intersection over Union (IoU) threshold for annotation.
  • --prompt_prefix: Prefix to add to every image generation prompt.
  • --prompt_suffix: Suffix to add to every image generation prompt, e.g., for adding details like resolution.
  • --negative_prompt: List of of negative prompts to guide the generation away from certain features.
  • --annotator_size: Size of the annotator model to use. Choose between base and large.

Copy link

github-actions bot commented Mar 24, 2024

☂️ Python Coverage

current status: ✅

Overall Coverage

Lines Covered Coverage Threshold Status
1046 518 50% 0% 🟢

New Files

No new covered files...

Modified Files

File Coverage Status
datadreamer/dataset_annotation/clip_annotator.py 55% 🟢
datadreamer/dataset_annotation/owlv2_annotator.py 67% 🟢
datadreamer/pipelines/generate_dataset_from_scratch.py 44% 🟢
datadreamer/prompt_generation/lm_synonym_generator.py 33% 🟢
datadreamer/prompt_generation/synonym_generator.py 85% 🟢
datadreamer/prompt_generation/wordnet_synonym_generator.py 82% 🟢
TOTAL 61% 🟢

updated for commit: b731e7e by action🐍

README.md Outdated Show resolved Hide resolved
Copy link

github-actions bot commented Mar 24, 2024

Test Results

  6 files    6 suites   50m 55s ⏱️
 90 tests  40 ✅  50 💤 0 ❌
540 runs  240 ✅ 300 💤 0 ❌

Results for commit b731e7e.

♻️ This comment has been updated with latest results.

Copy link
Contributor

@kozlov721 kozlov721 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@HonzaCuhel HonzaCuhel requested a review from sokovninn March 26, 2024 17:22
@sokovninn sokovninn merged commit 45e741d into dev Mar 26, 2024
9 checks passed
@sokovninn sokovninn deleted the feature/advanced-configuration branch April 8, 2024 20:28
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.

4 participants