isort

demo/demo.py

@@ -1,12 +1,13 @@
-import cv2
+import argparse
 import os
 import time
+
+import cv2
 import torch
-import argparse
-from nanodet.util import cfg, load_config, Logger
-from nanodet.model.arch import build_model
-from nanodet.util import load_model_weight
+
 from nanodet.data.transform import Pipeline
+from nanodet.model.arch import build_model
+from nanodet.util import Logger, cfg, load_config, load_model_weight
 from nanodet.util.path import mkdir
 
 image_ext = ['.jpg', '.jpeg', '.webp', '.bmp', '.png']

demo_mnn/python/demo_mnn.py

@@ -8,15 +8,14 @@
 '''
 
 import argparse
-from pathlib import Path
 from abc import ABCMeta, abstractmethod
+from pathlib import Path
 
 import cv2
-import numpy as np
 import matplotlib.pyplot as plt
-from tqdm import tqdm
+import numpy as np
 from scipy.special import softmax
-
+from tqdm import tqdm
 
 # Copy from nanodet/util/visualization.py
 _COLORS = np.array(

demo_ncnn/python/demo_ncnn.py

@@ -8,15 +8,14 @@
 '''
 
 import argparse
-from pathlib import Path
 from abc import ABCMeta, abstractmethod
+from pathlib import Path
 
 import cv2
-import numpy as np
 import matplotlib.pyplot as plt
-from tqdm import tqdm
+import numpy as np
 from scipy.special import softmax
-
+from tqdm import tqdm
 
 # Copy from nanodet/util/visualization.py
 _COLORS = np.array(
@@ -427,6 +426,7 @@ def infer_image(self, img_input):
 class NanoDetTorch(NanoDetABC):
     def __init__(self, model_path, cfg_path, *args, **kwargs):
         import torch
+
         from nanodet.model.arch import build_model
         from nanodet.util import Logger, cfg, load_config, load_model_weight
 

nanodet/data/collate.py

@@ -1,9 +1,21 @@
-import torch
-import torch.nn.functional as F
+# Copyright 2021 RangiLyu.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 
 import re
-from torch._six import container_abcs, string_classes, int_classes
 
+import torch
+from torch._six import container_abcs, int_classes, string_classes
 
 np_str_obj_array_pattern = re.compile(r'[SaUO]')
 

nanodet/data/dataset/__init__.py

@@ -1,4 +1,19 @@
+# Copyright 2021 RangiLyu.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 import copy
+
 from .coco import CocoDataset
 from .xml_dataset import XMLDataset
 

nanodet/data/dataset/base.py

@@ -1,7 +1,22 @@
+# Copyright 2021 RangiLyu.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 from abc import ABCMeta, abstractmethod
-import torch
+
 import numpy as np
 from torch.utils.data import Dataset
+
 from ..transform import Pipeline
 
 

nanodet/data/dataset/coco.py

@@ -1,8 +1,24 @@
+# Copyright 2021 RangiLyu.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 import os
-import torch
-import numpy as np
+
 import cv2
+import numpy as np
+import torch
 from pycocotools.coco import COCO
+
 from .base import BaseDataset
 
 

nanodet/data/dataset/xml_dataset.py

@@ -1,12 +1,25 @@
+# Copyright 2021 RangiLyu.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import logging
 import os
-import torch
-import numpy as np
-import cv2
 import time
-import logging
-from collections import defaultdict
 import xml.etree.ElementTree as ET
+from collections import defaultdict
+
 from pycocotools.coco import COCO
+
 from .coco import CocoDataset
 
 
@@ -21,7 +34,7 @@ def get_file_list(path, type='.xml'):
     return file_names
 
 
-class COCO_XML(COCO):
+class CocoXML(COCO):
 
     def __init__(self, annotation):
         """
@@ -121,7 +134,7 @@ def get_data_info(self, ann_path):
         ]
         """
         coco_dict = self.xml_to_coco(ann_path)
-        self.coco_api = COCO_XML(coco_dict)
+        self.coco_api = CocoXML(coco_dict)
         self.cat_ids = sorted(self.coco_api.getCatIds())
         self.cat2label = {cat_id: i for i, cat_id in enumerate(self.cat_ids)}
         self.cats = self.coco_api.loadCats(self.cat_ids)

nanodet/data/transform/__init__.py

@@ -1,2 +1,15 @@
+# Copyright 2021 RangiLyu.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 
-from .pipeline import Pipeline
+from .pipeline import Pipeline

nanodet/data/transform/color.py

@@ -1,7 +1,22 @@
-import numpy as np
-import cv2
+# Copyright 2021 RangiLyu.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 import random
 
+import cv2
+import numpy as np
+
 
 def random_brightness(img, delta):
     img += random.uniform(-delta, delta)

nanodet/data/transform/pipeline.py

@@ -1,7 +1,22 @@
-from .warp import warp_and_resize
-from .color import color_aug_and_norm
+# Copyright 2021 RangiLyu.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 import functools
 
+from .color import color_aug_and_norm
+from .warp import warp_and_resize
+
 
 class Pipeline:
     def __init__(self,

nanodet/data/transform/warp.py

@@ -1,14 +1,31 @@
+# Copyright 2021 RangiLyu.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import math
 import random
-import numpy as np
+
 import cv2
-import math
+import numpy as np
+
 
 def get_flip_matrix(prob=0.5):
     F = np.eye(3)
     if random.random() < prob:
         F[0, 0] = -1
     return F
 
+
 def get_perspective_matrix(perspective=0):
     """
 
@@ -36,8 +53,7 @@ def get_rotation_matrix(degree=0):
 def get_scale_matrix(ratio=(1, 1)):
     """
 
-    :param width_ratio:
-    :param height_ratio:
+    :param ratio:
     """
     Scl = np.eye(3)
     scale = random.uniform(*ratio)
@@ -114,6 +130,7 @@ def get_resize_matrix(raw_shape, dst_shape, keep_ratio):
         Rs[1, 1] *= d_h / r_h
         return Rs
 
+
 def warp_and_resize(meta, warp_kwargs, dst_shape, keep_ratio=True):
     # TODO: background, type
     raw_img = meta['img']
@@ -204,16 +221,3 @@ def warp_boxes(boxes, M, width, height):
 #         xy[:, [0, 2]] = xy[:, [0, 2]].clip(0, width)
 #         xy[:, [1, 3]] = xy[:, [1, 3]].clip(0, height)
 #         return xy
-
-
-
-
-
-
-
-
-
-
-
-
-

nanodet/evaluator/__init__.py

@@ -1,3 +1,17 @@
+# Copyright 2021 RangiLyu.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 from .coco_detection import CocoDetectionEvaluator
 
 

nanodet/evaluator/coco_detection.py

@@ -1,8 +1,22 @@
-import pycocotools.coco as coco
-from pycocotools.cocoeval import COCOeval
+# Copyright 2021 RangiLyu.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import copy
 import json
 import os
-import copy
+
+from pycocotools.cocoeval import COCOeval
 
 
 def xyxy2xywh(bbox):