^j<dZddlZddlZddlZddlZGddZy)z Real labels evaluator for ImageNet Paper: `Are we done with ImageNet?` - https://arxiv.org/abs/2006.07159 Based on Numpy example at https://github.com/google-research/reassessed-imagenet Hacked together by / Copyright 2020 Ross Wightman Nc"eZdZddZdZddZy)RealLabelsImagenetNc H|*t|5}tj|}dddnZtjt j t tjjddjd}tDcic]\}}d|dzdd|}}}||_ ||_ t|jt|jk(sJ||_|Dcic]}|gc}|_d|_y#1swYxYwcc}}wcc}w) N_infozimagenet_real_labels.jsonzutf-8ILSVRC2012_val_08dz.JPEGr)openjsonloadloadspkgutilget_data__name__ospathjoindecode enumerate real_labels filenameslentopk is_correct sample_idx)selfr real_jsonrrilabelsks `/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/timm/data/real_labels.py__init__zRealLabelsImagenet.__init__s  i 5K"ii 4  5 5**  277<<A\+]^eefmnpKS\]hSijiaQs 596Aj j&"4>>"c$*:*:&;;;; *./Q1b5/ 5 5 k 0sD D3 D Dc t|j}|j|ddd\}}|jj}|D]}|j|j }t jj|}|j|rX|jD]I}|j|jt|d|Dcgc]}||j|vc}K|xj dz c_ycc}w)NrT) maxrcpunumpyrrrrbasenamerrappendany) routputmaxk_ pred_batchpredfilenamer ps r! add_resultzRealLabelsImagenet.add_results499~ D!T48 :^^%++-  !D~~doo6Hww''1H)RAOOA&--d2AhOQ$"2"28"<<OPRR OOq O ! Ps Dc |G|jDcic]1}|ttj|j|dz3c}Sttj|j|dzScc}w)Nd)rfloatnpmeanr)rr s r! get_accuracyzRealLabelsImagenet.get_accuracy+se 9IMSAAuRWWT__Q%789C??S S!345; ;Ts6A8)N)r)N)r __module__ __qualname__r"r1r7r!rr s  !r>s"  "<"