apple / ml-cvnets

CVNets: A library for training computer vision networks
https://apple.github.io/ml-cvnets
Other
1.77k stars 225 forks source link

Confusion matrix is None. Check code Error #53

Closed HaoWuSR closed 1 year ago

HaoWuSR commented 1 year ago

Sorry to bother you. But I meet (Confusion matrix is None. Check code) Error

When I use the following command in ADE20K Dataset (However, it works well in PascalVoc2012): _export CFG_FILE="PATH_TO_CONFIG_FILE" export DEEPLABV3_MODEL_WEIGHTS="PATH_TO_MODEL_WEIGHTS" CUDA_VISIBLE_DEVICES=0 cvnets-eval-seg --common.config-file $CFG_FILE --common.results-loc seg_results --model.segmentation.pretrained $DEEPLABV3_MODEL_WEIGHTS --evaluation.segmentation.resize-input-images --evaluation.segmentation.mode validationset

It gets : Confusion matrix is None. Check code ......... acc_global, acc, iu, = conf_mat.compute() TypeError: cannot unpack non-iterable NoneType object

Maybe the error is caused by class ConfusionMatrix() with zero return value, I don't know how to figure it out Btw, the evaluation on a single image works properly.

HaoWuSR commented 1 year ago

I have solved it! sry!