How to train one's own dataset

Question

How to train one's own dataset

LL-XSJ opened this issue a year ago · 8 comments

Answer 1 · 2023-10-09T05:07:43.000Z

Hi, you should prepare your own dataset in COCO dataset format, then modified the path in these lines.

Answer 2 · 2023-10-09T05:31:17.000Z

Thank you for your reply. I did exactly that, but the training result has always been 0. I don't know why. Could you please reply to me

…

------------------ 原始邮件 ------------------ 发件人: "megvii-research/AnchorDETR" ***@***.***>; 发送时间: 2023年10月9日(星期一) 中午1:07 ***@***.***>; ***@***.******@***.***>; 主题: Re: [megvii-research/AnchorDETR] How to train one's own dataset (Issue #48) Hi, you should prepare your own dataset in COCO dataset format, then modified the path in these lines. — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>

Answer 3 · 2023-10-09T07:12:37.000Z

Your data may be not right. You may try to use this script to check it.

Answer 4 · 2023-10-09T08:07:32.000Z

My format is correct, I found anchor in the models folder_ After modifying the number of categories in detr.py, the results were all 0 after 50 rounds of training. Of course, I did not load the pre training weights, but I tested it and found that it is not related to whether to load the pre training weights. After I load the pre training weights, the training result is still 0. I guess it may be related to my modification of the number of categories. May I ask the author if I want to modify the anchor after changing the dataset_ What is the number of classes in detr. py?

…

------------------ 原始邮件 ------------------ 发件人: ***@***.***>; 发送时间: 2023年10月9日(星期一) 下午3:12 收件人: ***@***.***>; 抄送: ***@***.***>; ***@***.***>; 主题: Re: [megvii-research/AnchorDETR] How to train one's own dataset (Issue #48) Your data may be not right. You may try to use this script to check it. — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>

Answer 5 · 2023-10-09T08:25:49.000Z

How many rounds of training do I need to train my own dataset to converge without using the pre training weights given on the official website?

Answer 6 · 2023-10-10T03:29:12.000Z

Excuse me, is the pre training weight you provided on the official website the pre training weight for the entire anchor delta

Answer 7 · 2023-10-10T09:29:49.000Z

Hello, may I know how to modify the final category number of the model if I need to

Answer 8 · 2023-10-10T12:39:05.000Z

Hi, the final number of categories should be greater than the largest category index. But if the training result is always 0, in most cases the problem should be caused by the data format. Maybe you can try to use your data to train other detectors to check if the data is correct.