mirror of
https://github.com/alibaba/EasyCV.git
synced 2025-06-03 14:49:00 +08:00
654554cf65
Support Objects365 pretrain and Adding the DINO++ model can achieve an accuracy of 63.4mAP at a model scale of 200M(Under the same scale, the accuracy is the best)
11 lines
583 B
Python
11 lines
583 B
Python
#!/usr/bin/env python3
|
|
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
|
|
from .boxes import (batched_nms, bbox2result, bbox_overlaps, bboxes_iou,
|
|
box_cxcywh_to_xyxy, box_iou, box_xyxy_to_cxcywh,
|
|
distance2bbox, fp16_clamp, generalized_box_iou)
|
|
from .generator import MlvlPointGenerator
|
|
from .misc import (accuracy, gen_encoder_output_proposals,
|
|
gen_sineembed_for_position, interpolate, inverse_sigmoid,
|
|
select_single_mlvl)
|
|
from .postprocess import DetrPostProcess, output_postprocess, postprocess
|