Cum ștergi orice obiect dintr-o imagine cu ajutorul AI
Proiectul “Inpaint Anything” este un proiect open-source de IA în Python ce oferă capacitatea eliminării obiectelor nedorite din imagini. Utilizând tehnici de umplere a spațiului gol și de reconstrucție a texturii, acest proiect poate elimina obiecte dintr-o imagine și poate umple spațiul gol cu o reconstrucție realistă.
Nivelul de editare al imaginii ar trebui să fie superior celui oferit de un editor foto obișnuit în care selectarea măștii și “refacerea” imaginii era făcută de un editor uman. Dezvoltatorii proiectului oferă și un demo pe platforma huggingface pe care cel mai probabil îl veți putea folosi cu greu, mie mi-a luat cam 30 de minute să generez, rezultatul îl vedeți mai jos. Recunosc că nu am folosit o imagine foarte potrivită, aparent ai-ul doar a șters culorile și pinii de pe header-ul de la placa Radxa Zero și a lăsat doar bucata de plastic:
Instalare și utilizare
Mai indicat este să testați singuri acest proiect iar pentru aceasta va trebui să aveți instalat python3.8, o placă video decentă cu măcar 8Gb vRAM.
Instalare
python -m pip install torch torchvision torchaudio
python -m pip install -e segment_anything
python -m pip install -r lama/requirements.txt
Utilizare
python remove_anything.py \
--input_img ./example/remove-anything/dog.jpg \
--point_coords 200 450 \
--point_labels 1 \
--dilate_kernel_size 15 \
--output_dir ./results \
--sam_model_type "vit_h" \
--sam_ckpt ./pretrained_models/sam_vit_h_4b8939.pth \
--lama_config ./lama/configs/prediction/default.yaml \
--lama_ckpt ./pretrained_models/big-lama
Pentru mai multe detalii urmați instrucțiunile oferite de maintainerii proiectului pe pagina de github a acestora.
Mult succes!