CAMMA-public / rendezvous

A transformer-inspired neural network for surgical action triplet recognition from laparoscopic videos.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

HELP! A error when testing

csEylLee opened this issue · comments

Test weight used: ./checkpoint/run_2/rendezvous_l8_cholectcholect50_k0_batchnorm_lowres.pth
Traceback (most recent call last):
File "/home/cv5/rendezvous_code/pytorch/run_update.py", line 464, in
mAP_i = mAPi.compute_video_AP(ignore_null=set_chlg_eval)
File "/home/cv5/anaconda3/envs/torch110/lib/python3.8/site-packages/ivtmetrics/recognition.py", line 189, in compute_video_AP
classwise = average_precision_score(targets, predicts, average=None)
File "/home/cv5/anaconda3/envs/torch110/lib/python3.8/site-packages/sklearn/metrics/_ranking.py", line 234, in average_precision_score
return _average_binary_score(
File "/home/cv5/anaconda3/envs/torch110/lib/python3.8/site-packages/sklearn/metrics/_base.py", line 72, in _average_binary_score
raise ValueError("{0} format is not supported".format(y_type))
ValueError: multiclass-multioutput format is not supported

Are you using the CholecT50 dataloader ?

Are you using the CholecT50 dataloader ?

Yes. I've solved it now.