forked from mindspore-Ecosystem/mindspore
fix pylint warning
This commit is contained in:
parent
89eeb256dc
commit
6c412a94e3
|
@ -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({
|
||||||
|
|
Loading…
Reference in New Issue