[Initial version is here]

Please read below for the detail.

As I am not a professional engineer, I do not handle bug fixes or enhancements for this model or its features.

Special Thanks : Bing-su and Chat-GPT.

This is a yolo model for the detection penis.

The detection itself is performed using a square shape, similar to how faces are detected, because this model is trained based on yolov8n.pt. [This can be downloaded from draft version]

[Detected penis by penis_yolov8n.pt (draft version)]

Version 1.0 is another Yolo model that is segmented the target by polygon.

This was trained based on yolov8n-seg.pt. And also, I increased the number of training pictures more than 10%.

[Detected penis by penis_yolov8n_seg.pt (version 1.0)]

What do you use it for?

The purpose of using this model is not to swap the penis, but rather to exclude it.

When performing face swapping on an image like the one above using ADetailer, the penis gets cut off.

To resolve this issue, I enlisted the help of ChatGPT to create a mechanism that excludes the penis detected by the model from the face-swapping process.

As a result, I was able to achieve the following face-swapping effect.

[Example1: Face Swap by penis_yolov8n.pt (draft version)]

[Example2: Face Swap by penis_yolov8n.pt (draft version)]

[Example3: Face Swap by penis_yolov8n_seg.pt (version 1.0)]

[Example4: Face Swap by penis_yolov8n_seg.pt (version 1.0)]

[Example5: Face Swap by penis_yolov8n_seg.pt (version 1.0)]

Note :

Since penis_yolov8n performs detection using bounding boxes, the area around the mouth may sometimes appear distorted. While penis_yolov8n_seg offers higher accuracy, please be aware that it may still occasionally fail to produce the desired results.

[How to enable this model?]

For penis_yolov8n.pt, you can download it from here, and please rename the file as penis_yolov8n.pt.

For penis_yolo8n_seg.pt, you can download it from here, and please rename the file as penis-yolov8n_seg.pt.

Copy Yolo model file to <SD installation folder>\webui\models\adetailer

or if it did not detected

<SD installation folder>\webui\models\YOLO

[Modified ADetailer]

This is based on ADetailer 26.2.0.

ChatGPT has changed the code according to my requirements.

LICENSE

Regarding licensing, it is subject to the ADetailer license therefore distributed under the AGPL license.

[How to install modified ADetailer for penis-yolov8 model?]

Required Web UI : Automatic111 and Forge Classic

Please expand the zip file extension folder such as <SD installation folder>\webui\extensions

Manual PDF that is created by ChatGPT is available, but not checked the content by me.

(Both of English and Japanese version)

As reference information, the following pictures are my default setting.

If you want to disable this feature, please off all related options.

If this modified ADetailer is not fine for you, you can ask ChatGPT to modify furthermore.

[Tweet]

念願の顔入れ替え機能を作ってみました。

生成される画像によってはうまくいかないケースもあります。

また、口もとの画像と顔全体の皮膚感が違う事でかなり違和感を感じる場合もあります。

完璧では無いもののせっかく作ったのでこちらで公開させて頂きます。