what all changes do I need to make when to train with only person class.

[ssbilakeri]

I get bellow error while training with only person class

Wrong annotation: class_id = 1. But class_id should be [from 0 to 0], file: data/crowdhuman-608x608/273278,77a600008ca1359b.txt

Kindly suggest what should i do ?

[jkjung-avt]

1. In "data/gen_txts.py", remove the "head" ('hbox' -> class 0) code and redefine "full body" ('fbox') as class 0.

   yolov4_crowdhuman/data/gen_txts.py

   Lines 91 to 98 in 374b0e8

   	if 'hbox' in obj.keys(): # head
   	line = txt_line(0, obj['hbox'], img_w, img_h)
   	if line:
   	ftxt.write(line)
   	if 'fbox' in obj.keys(): # full body
   	line = txt_line(1, obj['fbox'], img_w, img_h)
   	if line:
   	ftxt.write(line)

2. Remove the "head" class and keep only the "person" class in data/crowdhuman.names.

3. Set "classes = 1" in data/crowdhuman-template.data.

[ssbilakeri]

Thank you very much.

…

On Mon, 9 Aug 2021, 4:01 pm JK Jung, ***@***.***> wrote: 1. In "data/gen_txts.py", remove the "head" ('hbox' -> class 0) code and redefine "full body" ('fbox') as class 0. https://github.com/jkjung-avt/yolov4_crowdhuman/blob/374b0e839e062d2039259ca18f3490f39cd122a8/data/gen_txts.py#L91-L98 2. Remove the "head" class and keep only the "person" class in data/crowdhuman.names <https://github.com/jkjung-avt/yolov4_crowdhuman/blob/374b0e839e062d2039259ca18f3490f39cd122a8/data/crowdhuman.names> . 3. Set "classes = 1" in data/crowdhuman-template.data <https://github.com/jkjung-avt/yolov4_crowdhuman/blob/374b0e839e062d2039259ca18f3490f39cd122a8/data/crowdhuman-template.data> . — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub <#26 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AOOYJBKZT3NPAEEH43NH3NTT36U7PANCNFSM5BSZFBCQ> . Triage notifications on the go with GitHub Mobile for iOS <https://apps.apple.com/app/apple-store/id1477376905?ct=notification-email&mt=8&pt=524675> or Android <https://play.google.com/store/apps/details?id=com.github.android&utm_campaign=notification-email> .