HRNet-Image-Classification icon indicating copy to clipboard operation
HRNet-Image-Classification copied to clipboard

how to get class label values & confusion matrix at test ? its not sufficient to get only accuracy..!

Open amitchougule121 opened this issue 6 years ago • 0 comments

I try with "HRNet-Image-Classification/lib/core/function

function.txt

.py" in this file, at line no. 104,

for i, (input, target) in enumerate(val_loader):
            output = model(input)
            batch_time.update(time.time() - end)
            target = target.cuda(non_blocking=True)
            loss = criterion(output, target)

in this I try with target variable, it shows correct values at train time, but at test time it shows the class label for image in which its stored. not accurate values. can anyone tell how to get class label values for "test.py" ? target variable shows inaccurate values at testing.

amitchougule121 avatar Dec 12 '19 10:12 amitchougule121