Commit Graph

19 Commits

Author SHA1 Message Date
Manuel Schmid 229ff81738
refactor: rename max_num_boxes to max_detections 2024-06-18 21:07:27 +02:00
Manuel Schmid 9c93c18d0b
fix: process other models than sam when using enhance 2024-06-16 21:05:10 +02:00
Manuel Schmid b585d9dfa7
fix: correctly count sam masks 2024-06-16 15:57:53 +02:00
Manuel Schmid f8f36828c7
feat: add disclaimer + skipping, stage2 won't properly work when used with inpaint or outpaint 2024-06-13 01:13:21 +02:00
Manuel Schmid b5b33361c9
feat: only process sam_options.dino_erode_or_dilate when set 2024-06-10 23:47:09 +02:00
Manuel Schmid 29967d3a18
feat: add dino_erode_or_dilate and dino_debug again 2024-06-10 23:23:38 +02:00
Manuel Schmid 757863c023
feat: wrap sam model in model patcher for predict 2024-06-10 22:42:35 +02:00
Manuel Schmid 651f9c5cfd
refactor: rename box_erode_or_dilate to dino_erode_or_dilate, add option dino_debug 2024-06-10 20:47:22 +02:00
Manuel Schmid 3873892b0a
feat: change default_inpaint_mask_sam_model to match sam model registry 2024-06-10 20:45:56 +02:00
Manuel Schmid ce1fb74270
feat: add download for sam models to config 2024-06-10 20:33:49 +02:00
Manuel Schmid 8a81993940
wip: remove ultralytics, always use manual sam for image mask instead of rembg 2024-06-10 01:33:03 +02:00
Manuel Schmid 09e23f5509
refactor: add info to inpaint_mask_sam_prompt_text, rename variable for dino debugging 2024-06-09 23:09:46 +02:00
Manuel Schmid 57c049858c
feat: add debug dino and mask dilate and erode 2024-06-09 22:31:41 +02:00
Manuel Schmid ff9fa6c837
feat: add box_erode_or_dilate to generate_mask_from_image, expose more sam return values 2024-06-09 18:45:13 +02:00
rayronvictor dd5264dd14 fix error when run_grounded_sam find no boxes 2024-01-26 07:58:57 -03:00
Manuel Schmid 90be73a6df
feat: add model patching
automatically unload model when not needed anymore
2024-01-26 02:05:38 +01:00
rayronvictor 228a0aaeea Add SAM support 2024-01-25 16:45:02 -03:00
rayronvictor f0fb9783e1 Add options to u2net_cloth_seg 2024-01-24 21:26:23 -03:00
Manuel Schmid db5512fb93
feat: add inpaint mask generation functionality
uses rembg, see https://github.com/danielgatis/rembg
2024-01-24 02:22:50 +01:00