Yizhao-AwakeAI comments

Repositories
Issues
Comments

Results 1 comments of


                                            Yizhao-AwakeAI

Fine Tune a SpatioTemporal Action Detection model on a custom dataset in AVA format

change the mmaction/models/roi_heads/bbox_heads/bbox_head.py. Add these 2 lines after the row 244, which can change the gt's class number from 81 to 7 for sampling_result in sampling_results: sampling_result.pos_gt_labels = sampling_result.pos_gt_labels[:, :self.num_classes]