README.md

metadata

datasets:
  - deepghs/real_person_detection
  - deepghs/anime_person_detection
pipeline_tag: object-detection
library_name: dghs-imgutils
tags:
  - art
  - anime
  - photo
  - person

These models are trained on deepghs/anime_person_detection and open-sourced real photos datasets.

So both anime and photos are supported.

Use this with dghs-realutils

pip install dghs-realutils

from realutils.detect import detect_persons

print(detect_persons('yolo/solo.jpg'))
#[((0, 30, 398, 599), 'person', 0.926707923412323)]
print(detect_persons('yolo/2girls.jpg'))
# [((0, 74, 760, 1598), 'person', 0.7578195333480835), ((437, 33, 1200, 1600), 'person', 0.6875205039978027)]
print(detect_persons('yolo/3+cosplay.jpg'))
# [((106, 69, 347, 591), 'person', 0.8794167041778564), ((326, 14, 592, 534), 'person', 0.8018194437026978), ((167, 195, 676, 675), 'person', 0.5351650714874268)]
print(detect_persons('yolo/multiple.jpg'))
# [((1305, 441, 1891, 1534), 'person', 0.8789498805999756), ((206, 191, 932, 1533), 'person', 0.8423126935958862), ((1054, 170, 1417, 1055), 'person', 0.8138357996940613), ((697, 659, 1473, 1534), 'person', 0.7926754951477051), ((685, 247, 1128, 1526), 'person', 0.5261526703834534), ((690, 251, 1125, 1126), 'person', 0.4193646311759949)]

For more information, see documentation of realutils.

Model	Type	FLOPS	Params	F1 Score	Threshold	precision(B)	recall(B)	mAP50(B)	mAP50-95(B)	F1 Plot	Confusion	Labels
person_detect_v0_l_yv11	yolo	87.3G	25.3M	0.79	0.359	0.84037	0.74055	0.82796	0.57272	plot	confusion	person
person_detect_v0_m_yv11	yolo	68.2G	20.1M	0.78	0.351	0.83393	0.73614	0.82195	0.56267	plot	confusion	person
person_detect_v0_s_yv11	yolo	21.5G	9.43M	0.75	0.344	0.82356	0.6967	0.79224	0.52304	plot	confusion	person
person_detect_v0_n_yv11	yolo	6.44G	2.59M	0.71	0.325	0.80096	0.64148	0.74612	0.46875	plot	confusion	person
person_detect_v0_l	yolo	165G	43.6M	0.79	0.359	0.83674	0.74182	0.82536	0.57022	plot	confusion	person
person_detect_v0_m	yolo	79.1G	25.9M	0.78	0.363	0.83439	0.72529	0.81314	0.55388	plot	confusion	person
person_detect_v0_s	yolo	28.6G	11.1M	0.76	0.346	0.82522	0.69696	0.79105	0.52201	plot	confusion	person
person_detect_v0_n	yolo	8.19G	3.01M	0.72	0.32	0.80883	0.64552	0.74996	0.47272	plot	confusion	person