Update some file headers.

Author: rwightman

effdet/data/dataset.py

@@ -1,4 +1,4 @@
-""" COCO dataset (quick and dirty)
+""" Detection dataset
 
 Hacked together by Ross Wightman
 """
@@ -9,7 +9,7 @@
 
 
 class DetectionDatset(data.Dataset):
-    """`MS Coco Detection <http://mscoco.org/dataset/#detections-challenge2016>`_ Dataset.
+    """`Object Detection Dataset. Use with parsers for COCO, VOC, and OpenImages.
     Args:
         parser (string, Parser):
         transform (callable, optional): A function/transform that  takes in an PIL image

effdet/data/dataset_config.py

@@ -1,5 +1,8 @@
-import os
+""" COCO, VOC, OpenImages dataset configurations
 
+Copyright 2020 Ross Wightman
+"""
+import os
 from dataclasses import dataclass, field
 from typing import Dict
 

effdet/data/dataset_factory.py

@@ -1,3 +1,7 @@
+""" Dataset factory
+
+Copyright 2020 Ross Wightman
+"""
 import os
 from pathlib import Path
 

effdet/data/loader.py

@@ -1,6 +1,6 @@
 """ Object detection loader/collate
 
-Hacked together by Ross Wightman
+Hacked together by / Copyright 2020 Ross Wightman
 """
 import torch.utils.data
 from .transforms import *

effdet/data/parsers/parser_coco.py

@@ -1,3 +1,7 @@
+""" COCO dataset parser
+
+Copyright 2020 Ross Wightman
+"""
 import os
 import numpy as np
 from pycocotools.coco import COCO

effdet/data/parsers/parser_config.py

@@ -1,3 +1,7 @@
+""" Dataset parser configs
+
+Copyright 2020 Ross Wightman
+"""
 from dataclasses import dataclass
 
 __all__ = ['CocoParserCfg', 'OpenImagesParserCfg', 'VocParserCfg']

effdet/data/parsers/parser_factory.py

@@ -1,3 +1,7 @@
+""" Parser factory
+
+Copyright 2020 Ross Wightman
+"""
 from .parser_coco import CocoParser
 from .parser_open_images import OpenImagesParser
 from .parser_voc import VocParser

effdet/data/parsers/parser_open_images.py

@@ -1,3 +1,7 @@
+""" OpenImages dataset parser
+
+Copyright 2020 Ross Wightman
+"""
 import numpy as np
 import pandas as pd
 import os

effdet/data/parsers/parser_voc.py

@@ -1,3 +1,7 @@
+""" Pascal VOC dataset parser
+
+Copyright 2020 Ross Wightman
+"""
 import os
 import xml.etree.ElementTree as ET
 from collections import defaultdict

effdet/data/random_erasing.py

@@ -1,3 +1,7 @@
+""" Multi-Scale RandomErasing
+
+Copyright 2020 Ross Wightman
+"""
 import random
 import math
 import torch

effdet/loss.py

@@ -1,3 +1,10 @@
+""" EfficientDet Focal, Huber/Smooth L1 loss fns w/ jit support
+
+Based on loss fn in Google's automl EfficientDet repository (Apache 2.0 license).
+https://github.com/google/automl/tree/master/efficientdet
+
+Copyright 2020 Ross Wightman
+"""
 import torch
 import torch.nn as nn
 import torch.nn.functional as F