feat(YOLOX): add header

yolox/core/__init__.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
 
 from .launch import launch
 from .trainer import Trainer

yolox/core/launch.py

@@ -5,7 +5,6 @@
 # Copyright (c) Facebook, Inc. and its affiliates.
 # Copyright (c) Megvii, Inc. and its affiliates.
 
-
 from loguru import logger
 
 import torch

yolox/core/trainer.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
+
 import datetime
 import os
 import time

yolox/data/__init__.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
 
 from .data_augment import TrainTransform, ValTransform
 from .data_prefetcher import DataPrefetcher

yolox/data/data_augment.py

@@ -1,3 +1,6 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
 """
 Data augmentation functionality. Passed as callable transformations to
 Dataset classes.

yolox/data/data_prefetcher.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
 
 import random
 

yolox/data/dataloading.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
+
 import os
 import random
 

yolox/data/datasets/__init__.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
 
 from .coco import COCODataset
 from .coco_classes import COCO_CLASSES

yolox/data/datasets/coco.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
+
 import os
 
 import cv2

yolox/data/datasets/coco_classes.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
-
+# Copyright (c) Megvii, Inc. and its affiliates.
 
 COCO_CLASSES = (
     "person",

yolox/data/datasets/datasets_wrapper.py

@@ -1,5 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
+
 import bisect
 from functools import wraps
 

yolox/data/datasets/mosaicdetection.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
+
 import random
 
 import cv2

yolox/data/datasets/voc.py

@@ -1,10 +1,10 @@
-"""VOC Dataset Classes
-
-Original author: Francisco Massa
-https://github.com/fmassa/vision/blob/voc_dataset/torchvision/datasets/voc.py
-
-Updated by: Ellis Brown, Max deGroot
-"""
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Code are based on
+# https://github.com/fmassa/vision/blob/voc_dataset/torchvision/datasets/voc.py
+# Copyright (c) Francisco Massa.
+# Copyright (c) Ellis Brown, Max deGroot.
+# Copyright (c) Megvii, Inc. and its affiliates.
 
 import os
 import os.path

yolox/data/datasets/voc_classes.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
-
+# Copyright (c) Megvii, Inc. and its affiliates.
 
 # VOC_CLASSES = ( '__background__', # always index 0
 VOC_CLASSES = (

yolox/data/samplers.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
 
 import itertools
 from typing import Optional

yolox/evalutors/__init__.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
 
 from .coco_evaluator import COCOEvaluator

yolox/evalutors/coco_evaluator.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
+
 import contextlib
 import io
 import itertools
@@ -87,7 +91,7 @@ class COCOEvaluator:
             model_trt.load_state_dict(torch.load(trt_file))
 
             x = torch.ones(1, 3, test_size[0], test_size[1]).cuda()
-            dump_out = model(x)
+            model(x)
             model = model_trt
 
         for cur_iter, (imgs, _, info_imgs, ids) in enumerate(

yolox/evalutors/voc_eval.py

@@ -1,8 +1,9 @@
-# --------------------------------------------------------
-# Fast/er R-CNN
-# Licensed under The MIT License [see LICENSE for details]
-# Written by Bharath Hariharan
-# --------------------------------------------------------
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Code are based on
+# https://github.com/rbgirshick/py-faster-rcnn/blob/master/lib/datasets/voc_eval.py
+# Copyright (c) Bharath Hariharan.
+# Copyright (c) Megvii, Inc. and its affiliates.
 
 import os
 import pickle

yolox/evalutors/voc_evaluator.py

@@ -1,3 +1,8 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) Megvii, Inc. and its affiliates.
+
+# NOTE: this file is not finished.
 import sys
 import tempfile
 import time
@@ -5,12 +10,9 @@ from tqdm import tqdm
 
 import torch
 
-# TODO check VOC
 from yolox.data.dataset.vocdataset import ValTransform
 from yolox.utils import get_rank, is_main_process, make_pred_vis, make_vis, synchronize
 
-# TODO refactor this file in the future.
-
 
 def _accumulate_predictions_from_multiple_gpus(predictions_per_gpu):
     all_predictions = dist.scatter_gather(predictions_per_gpu)

yolox/exp/__init__.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 from .base_exp import BaseExp
 from .build import get_exp

yolox/exp/base_exp.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
 # Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
+
 import ast
 import pprint
 from abc import ABCMeta, abstractmethod

yolox/exp/build.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 import importlib
 import os

yolox/layers/__init__.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 from .fast_coco_eval_api import COCOeval_opt

yolox/layers/fast_coco_eval_api.py

@@ -3,6 +3,7 @@
 # This file comes from
 # https://github.com/facebookresearch/detectron2/blob/master/detectron2/evaluation/fast_eval_api.py
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 import copy
 import time

yolox/models/__init__.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 from .darknet import CSPDarknet, Darknet
 from .losses import IOUloss

yolox/models/darknet.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python
 # -*- encoding: utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 from torch import nn
 

yolox/models/losses.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
+
 import torch
 import torch.nn as nn
 

yolox/models/network_blocks.py

@@ -1,9 +1,13 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
+
 import torch
 import torch.nn as nn
 
 
 class SiLU(nn.Module):
-    # export-friendly version of nn.SiLU()
+    """export-friendly version of nn.SiLU()"""
 
     @staticmethod
     def forward(x):
@@ -23,9 +27,7 @@ def get_activation(name="silu", inplace=True):
 
 
 class BaseConv(nn.Module):
-    """
-    A Conv2d -> Batchnorm -> silu/leaky relu block
-    """
+    """A Conv2d -> Batchnorm -> silu/leaky relu block"""
 
     def __init__(self, in_channels, out_channels, ksize, stride, groups=1, bias=False, act="silu"):
         super().__init__()
@@ -99,7 +101,7 @@ class ResLayer(nn.Module):
 
 
 class SPPBottleneck(nn.Module):
-    # Spatial pyramid pooling layer used in YOLOv3-SPP
+    """Spatial pyramid pooling layer used in YOLOv3-SPP"""
     def __init__(self, in_channels, out_channels, kernel_sizes=(5, 9, 13), activation="silu"):
         super().__init__()
         hidden_channels = in_channels // 2
@@ -126,6 +128,8 @@ class CSPLayer(nn.Module):
     ):
         """
         Args:
+            in_channels (int): input channels.
+            out_channels (int): output channels.
             n (int): number of Bottlenecks. Default value: 1.
         """
         # ch_in, ch_out, number, shortcut, groups, expansion
@@ -150,6 +154,7 @@ class CSPLayer(nn.Module):
 
 class Focus(nn.Module):
     """Focus width and height information into channel space."""
+
     def __init__(self, in_channels, out_channels, ksize=1, stride=1):
         super().__init__()
         self.conv = BaseConv(in_channels * 4, out_channels, ksize, stride)

yolox/models/yolo_fpn.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
+
 import torch
 import torch.nn as nn
 

yolox/models/yolo_head.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 import math
 from loguru import logger

yolox/models/yolo_pafpn.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
+
 import torch
 import torch.nn as nn
 

yolox/models/yolox.py

@@ -1,3 +1,6 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 import torch.nn as nn
 

yolox/utils/__init__.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 from .allreduce_norm import *
 from .boxes import *

yolox/utils/allreduce_norm.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 import pickle
 from collections import OrderedDict

yolox/utils/boxes.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
+
 import numpy as np
 
 import torch

yolox/utils/checkpoint.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 import os
 import shutil
 from loguru import logger

yolox/utils/dist.py

@@ -1,8 +1,9 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
-# This file comes from
+# This file mainly comes from
 # https://github.com/facebookresearch/detectron2/blob/master/detectron2/utils/comm.py
 # Copyright (c) Facebook, Inc. and its affiliates.
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 """
 This file contains primitives for multi-gpu communication.
 This is useful when doing distributed training.
@@ -248,7 +249,7 @@ def shared_random_seed():
 
 
 def time_synchronized():
-    # pytorch-accurate time
+    """pytorch-accurate time"""
     if torch.cuda.is_available():
         torch.cuda.synchronize()
     return time.time()

yolox/utils/ema.py

@@ -1,3 +1,6 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 import math
 from copy import deepcopy
 

yolox/utils/logger.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
+
 import inspect
 import os
 import sys

yolox/utils/lr_scheduler.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
+
 import math
 from functools import partial
 
@@ -7,10 +11,14 @@ class LRScheduler:
         """
         Supported lr schedulers: [cos, warmcos, multistep]
 
-        Extra keyword arguments:
-          - cos: None
-          - warmcos: [warmup_epochs, warmup_lr_start (default 1e-6)]
-          - multistep: [milestones (epochs), gamma (default 0.1)]
+        Args:
+            lr (float): learning rate.
+            iters_per_peoch (int): number of iterations in one epoch.
+            total_epochs (int): number of epochs in training.
+            kwargs (dict):
+                - cos: None
+                - warmcos: [warmup_epochs, warmup_lr_start (default 1e-6)]
+                - multistep: [milestones (epochs), gamma (default 0.1)]
         """
 
         self.lr = lr

yolox/utils/model_utils.py

@@ -1,5 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
+
 from copy import deepcopy
 
 import torch

yolox/utils/visualize.py

@@ -1,4 +1,6 @@
-# -*- coding: utf-8 -*-
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 import cv2
 import numpy as np