hello_cc

以COCO数据集为例，通过get_dataset获取Dataset的信息(继承COCO和CTDetDataset类)。再根据update_dataset_info_and_set_heads 设置对应的head。 ctdet对应的head有三个：hm(heatmap)，wh(边…

1. Root类 对应绿色框的aggregation node,有多个输入对象，用于聚合各个层的信息。 2. Tree类 对应红色框的hierarchical deep agrregation(HDA)。其中主要包括几个核心部分: level=1时，self.tree1和sel…

首先是在CtdetTrainer(继承BaseTrainer)的init里初始化loss, 在ExdetLoss类获取loss。 假设输入图像大小为$HWC$, 经过model之后，分别接三个head获取对应的特征。再加不同的损失函数进行约束： 1. hm head &#x26…