| @@ -17,15 +17,15 @@ | |||||
| from __future__ import division | from __future__ import division | ||||
| import os | import os | ||||
| import mmcv | |||||
| import cv2 | |||||
| import numpy as np | import numpy as np | ||||
| from numpy import random | from numpy import random | ||||
| import mmcv | |||||
| import mindspore.dataset as de | import mindspore.dataset as de | ||||
| import mindspore.dataset.vision.c_transforms as C | import mindspore.dataset.vision.c_transforms as C | ||||
| from mindspore.mindrecord import FileWriter | from mindspore.mindrecord import FileWriter | ||||
| from src.config import config | from src.config import config | ||||
| import cv2 | |||||
| def bbox_overlaps(bboxes1, bboxes2, mode='iou'): | def bbox_overlaps(bboxes1, bboxes2, mode='iou'): | ||||
| """Calculate the ious between each bbox of bboxes1 and bboxes2. | """Calculate the ious between each bbox of bboxes1 and bboxes2. | ||||
| @@ -14,12 +14,11 @@ | |||||
| # ============================================================================ | # ============================================================================ | ||||
| """coco eval for maskrcnn_mobilenetv1""" | """coco eval for maskrcnn_mobilenetv1""" | ||||
| import json | import json | ||||
| import mmcv | |||||
| import numpy as np | import numpy as np | ||||
| from pycocotools.coco import COCO | from pycocotools.coco import COCO | ||||
| from pycocotools.cocoeval import COCOeval | from pycocotools.cocoeval import COCOeval | ||||
| from pycocotools import mask as maskUtils | from pycocotools import mask as maskUtils | ||||
| import mmcv | |||||
| from src.config import config | from src.config import config | ||||
| _init_value = np.array(0.0) | _init_value = np.array(0.0) | ||||
| @@ -15,8 +15,8 @@ | |||||
| """ | """ | ||||
| network config setting, will be used in dataset.py, run_general_distill.py and run_task_distill.py | network config setting, will be used in dataset.py, run_general_distill.py and run_task_distill.py | ||||
| """ | """ | ||||
| import mindspore.common.dtype as mstype | |||||
| from easydict import EasyDict as edict | from easydict import EasyDict as edict | ||||
| import mindspore.common.dtype as mstype | |||||
| from .tinybert_model import BertConfig | from .tinybert_model import BertConfig | ||||
| common_cfg = edict({ | common_cfg = edict({ | ||||
| @@ -15,8 +15,8 @@ | |||||
| """config script for task distill""" | """config script for task distill""" | ||||
| import mindspore.common.dtype as mstype | |||||
| from easydict import EasyDict as edict | from easydict import EasyDict as edict | ||||
| import mindspore.common.dtype as mstype | |||||
| from .tinybert_model import BertConfig | from .tinybert_model import BertConfig | ||||
| phase1_cfg = edict({ | phase1_cfg = edict({ | ||||