o ߥi@@srddlZddlZddlZidddddddd d d d d ddddddddddddddddddd d!d"d#id$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEidFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgidhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~dddddddddddiddddddddddddddddddddddddddddddddddiddddddddddddddddddddddddÓddœddǓddɓdd˓dd͓iddϓddѓddӓddՓddדddٓddۓddݓddߓdddddddddddddddddddddddZddeDZddeDZgdZddZ ddZ ddZ ddZ ddZ dS(Nwall1ceiling2door3stair4ladder5 escalator6Playground_slide7handrail_or_fence8window9others0rail10goal11pillar12pole13floor14ground15grass16sand17athletic_field18road19path20 crosswalk21building22house23bridge24tower25windmill26well_or_well_lid27other_construction28sky29mountain30stone31wood32ice33 snowfield34 grandstand35sea36river37lake38 waterfall39water40billboard_or_Bulletin_Board41 sculpture42pipeline43flag44parasol_or_umbrella45cushion_or_carpet46tent47 roadblock48car49bus50truck51bicycle52 motorcycle53wheeled_machine54 ship_or_boat55raft56airplane57tyre58 traffic_light59lamp60person61cat62dog63horse64cattle65 other_animal66tree67flower68 other_plant69toy70ball_net71 backboard72 skateboard73bat74ball75$cupboard_or_showcase_or_storage_rack76box77traveling_case_or_trolley_case78basket79bag_or_package80 trash_can81cage82plate83tub_or_bowl_or_pot84 bottle_or_cup85barrel86fishbowl87bed88pillow89 table_or_desk90 chair_or_seat91bench92sofa93shelf94bathtub95gun96commode97roaster98 other_machine99 refrigerator100washing_machine101Microwave_oven102fan103curtain104textiles105clothes106painting_or_poster107mirror108flower_pot_or_vase109clock110book111tool112 blackboard113tissue114screen_or_television115computer116printer117 Mobile_phone118119120121122123124)keyboardother_electronic_productfruitfood instrumenttraincCsi|] \}}|t|qSint.0kvrro/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/cv/video_panoptic_segmentation/visualizer.py rcCsi|] \}}t||qSrrrrrrrr))r(r)@r)sidewalk)#)r.)Frr)r)fr)fence)r)r)rrr)z traffic light))z traffic sign)rr) vegetation)kr )terrain )r#)r< )r)r| )r<)rider )rr)rd )rrr )rh)rrr)rf)rr*d)r)rPr1)rl)rr)rj)wr( )voidrrr)r9r-r;cCs.tt|d}|dd}t|dS)Nzutf-8ir2)hashlibsha256strencode hexdigestr)numhexrrr sha256nums  rCcCst|tjr1|}tt|jdg}tj|tjd}t dD]}|d|d|f<|d}q |Sg}t dD] }| |d|d}q7|S)Nr)dtype.) isinstancenpndarraycopytuplelistshapezerosuint8rangeappend)id_map id_map_copy rgb_shapergb_mapicolor_rrrid2rgbs     rXcCsRt|tj}|djddd}tD]}||dk}|r&|d||<q|S)N.Nraxisr r )rG zeros_likeastyperNrepeat city_labelsany)cat_map color_map each_classindexrrrcityscapes_cat2rgbs  rfcCsVt|tj}|djddd}t|D]}|dkrqtt||||k<q|S)NrYrrZr[r)rGr]r^rNr_uniquerXrC) track_maprcid_currrr trackmap2rgbsrjcCsdt|jdD](}tj|t||dt||dft||dt||dfdddq|S)Nrr r r)rrr-) thickness)rOrLcv2 rectangler)vis_imgbboxesrerrrdraw_bbox_on_imgs  rp)r<rlnumpyrG vip_seg_labelitemsvip_seg_label_to_idvip_seg_id_to_labelr`rCrXrfrjrprrrrs      !"#$%&'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\]^_`abcdefghijklmnopqrstuvwx