You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I've changed custom_detection.yml to have
num_classes: 2 # your dataset classes
remap_mscoco_category: False
I'm trying to train on custom data with one class. I train using this command: $env:CUDA_VISIBLE_DEVICES="0"; python train.py -c configs/dfine/custom/objects365/dfine_hgnetv2_s_obj2custom.yml --use-amp --seed=0 -t dfine_s_obj2coco.pth.
When I export the model to onnx, it outputs confidence scores for 300 classes. Is this expected or am I missing something in my training?
The text was updated successfully, but these errors were encountered:
I've changed custom_detection.yml to have
num_classes: 2 # your dataset classes
remap_mscoco_category: False
I'm trying to train on custom data with one class. I train using this command: $env:CUDA_VISIBLE_DEVICES="0"; python train.py
-c configs/dfine/custom/objects365/dfine_hgnetv2_s_obj2custom.yml
--use-amp --seed=0 -t dfine_s_obj2coco.pth.When I export the model to onnx, it outputs confidence scores for 300 classes. Is this expected or am I missing something in my training?
The text was updated successfully, but these errors were encountered: