Browse Source

fix pylint warning

pull/14760/head
w00449189 4 years ago
parent
commit
6c412a94e3
4 changed files with 6 additions and 7 deletions
  1. +3
    -3
      model_zoo/official/cv/maskrcnn_mobilenetv1/src/dataset.py
  2. +1
    -2
      model_zoo/official/cv/maskrcnn_mobilenetv1/src/util.py
  3. +1
    -1
      model_zoo/official/nlp/tinybert/src/gd_config.py
  4. +1
    -1
      model_zoo/official/nlp/tinybert/src/td_config.py

+ 3
- 3
model_zoo/official/cv/maskrcnn_mobilenetv1/src/dataset.py View File

@@ -17,15 +17,15 @@
from __future__ import division

import os
import mmcv
import cv2
import numpy as np
from numpy import random

import mmcv
import mindspore.dataset as de
import mindspore.dataset.vision.c_transforms as C
from mindspore.mindrecord import FileWriter
from src.config import config
import cv2

def bbox_overlaps(bboxes1, bboxes2, mode='iou'):
"""Calculate the ious between each bbox of bboxes1 and bboxes2.


+ 1
- 2
model_zoo/official/cv/maskrcnn_mobilenetv1/src/util.py View File

@@ -14,12 +14,11 @@
# ============================================================================
"""coco eval for maskrcnn_mobilenetv1"""
import json
import mmcv
import numpy as np
from pycocotools.coco import COCO
from pycocotools.cocoeval import COCOeval
from pycocotools import mask as maskUtils
import mmcv

from src.config import config

_init_value = np.array(0.0)


+ 1
- 1
model_zoo/official/nlp/tinybert/src/gd_config.py View File

@@ -15,8 +15,8 @@
"""
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
import mindspore.common.dtype as mstype
from .tinybert_model import BertConfig

common_cfg = edict({


+ 1
- 1
model_zoo/official/nlp/tinybert/src/td_config.py View File

@@ -15,8 +15,8 @@

"""config script for task distill"""

import mindspore.common.dtype as mstype
from easydict import EasyDict as edict
import mindspore.common.dtype as mstype
from .tinybert_model import BertConfig

phase1_cfg = edict({


Loading…
Cancel
Save