lightweight的prepare_train_labels
prepare_train_labels建立了俩个字典首先需要了解到cocokeypoint数据的内容。标住都存于annotations,每个标注都对应一个图片,iscrow=1表示人群拥挤并未标住。因此代码建立了一个字典,字典的键值是图片的编号,对应俩个列表第一个列表存的是标注的关键点的信息,第二个列表是图片未标注的部分信息。
接下来它就根据图片中标记的人做循环,每个人都可以作为prepared_annotation的keypoints一次,然后在其他的prepared_annotation出现在prepared_annotation。

浙公网安备 33010602011771号