lichunxiang93
Results
1
comments of
lichunxiang93
我看了torchvision中的源码，源码中dataset会根据数据集中是否有keypoints和mask获得对应的标签字典，我在想是否可以直接在您代码上再加入一个keypoint检测分支即可，源码中会根据mask和keypoint分支类型的实现计算得到loss字典。正在尝试该方法......