[MaskRCNN] move dataset code to dataset/

examples/FasterRCNN/NOTES.md

 
 ### File Structure
 This is a minimal implementation that simply contains these files:
-+ dataset.py: the dataset interface
-+ coco.py: load COCO data to the dataset interface
-+ data.py: prepare data for training & inference
-+ common.py: common data preparation utilities
++ train.py,predict.py: main entry script
 + modeling/generalized_rcnn.py: implement variants of generalized R-CNN architecture
 + modeling/backbone.py: implement backbones
 + modeling/model_{fpn,rpn,frcnn,mrcnn,cascade}.py: implement FPN,RPN,Fast/Mask/Cascade R-CNN models.
 + modeling/model_box.py: implement box-related symbolic functions
-+ train.py: main entry script
++ dataset/dataset.py: the dataset interface
++ dataset/coco.py: load COCO data to the dataset interface
++ data.py: prepare data for training & inference
++ common.py: common data preparation utilities
 + utils/: third-party helper functions
 + eval.py: evaluation utilities
 + viz.py: visualization utilities

examples/FasterRCNN/dataset/__init__.py

+from .dataset import *
+from .coco import *

examples/FasterRCNN/predict.py

@@ -17,11 +17,10 @@ from tensorpack.predict import MultiTowerOfflinePredictor, OfflinePredictor, Pre
 from tensorpack.tfutils import get_model_loader, get_tf_version_tuple
 from tensorpack.utils import fs, logger
 
-from coco import register_coco
+from dataset import DatasetRegistry, register_coco
 from config import config as cfg
 from config import finalize_configs
 from data import get_eval_dataflow, get_train_dataflow
-from dataset import DatasetRegistry
 from eval import DetectionResult, multithread_predict_dataflow, predict_image
 from modeling.generalized_rcnn import ResNetC4Model, ResNetFPNModel
 from viz import draw_annotation, draw_final_outputs, draw_predictions, draw_proposal_recall

examples/FasterRCNN/train.py

@@ -10,7 +10,7 @@ from tensorpack import *
 from tensorpack.tfutils import collect_env_info
 from tensorpack.tfutils.common import get_tf_version_tuple
 
-from coco import register_coco
+from dataset import register_coco
 from config import config as cfg
 from config import finalize_configs
 from data import get_train_dataflow

examples/tox.ini

@@ -2,4 +2,5 @@
 max-line-length = 120
 ignore = F403,F405,E402,E741,E742,E743,W504,W605
 exclude = private,
-          FasterRCNN/utils
+          FasterRCNN/utils,
+					__init__.py