TAO-Dataset / tao

Code for downloading and using the TAO dataset: http://taodataset.org/
Other
194 stars 32 forks source link

imagenetvid iou #31

Closed hf1995 closed 2 years ago

hf1995 commented 2 years ago

_numtotal has duplicate statistics. I wonder if line 67 https://github.com/TAO-Dataset/tao/blob/3282f29212a40ddc3785cd4d4cd311c6b7b8e084/tao/toolkit/tao/eval.py#L67 should be deleted, and _numtotal can be calculated on line 70 https://github.com/TAO-Dataset/tao/blob/3282f29212a40ddc3785cd4d4cd311c6b7b8e084/tao/toolkit/tao/eval.py#L70.

achalddave commented 2 years ago

Great catch, yes! If you could submit a PR, happy to merge. Otherwise, I'll fix in the next release / update. I'll leave this issue open until then.

Edit: Nvm, fixed!