D-FINE icon indicating copy to clipboard operation
D-FINE copied to clipboard

dataset train

Open gyniy opened this issue 11 months ago • 6 comments

Problems encountered when training with a custom coco dataset,Dimension box format is (xmin,ymin,w,h)

Image

gyniy avatar Jun 08 '25 07:06 gyniy

你解决这个问题了吗?

Donaghys avatar Jun 09 '25 01:06 Donaghys

我训了20多个epoch后指标,就开始缓慢上涨了

gyniy avatar Jun 09 '25 05:06 gyniy

能发下训练的指令吗?还有数据集的类别是几个?

Donaghys avatar Jun 09 '25 06:06 Donaghys

训练指令CUDA_VISIBLE_DEVICES=0 torchrun --master_port=7777 --nproc_per_node=1 train.py -c configs/dfine/dfine_hgnetv2_${model}_coco.yml --use-amp --seed=0,我的数据集的类别是9

gyniy avatar Jun 09 '25 06:06 gyniy

你训练的类别都在coco里面吗?

Donaghys avatar Jun 12 '25 06:06 Donaghys

Hi @gyniy , Have you solved the problem?

edwardnguyen1705 avatar Jun 20 '25 08:06 edwardnguyen1705