A notebook demo can be found in [demo/inference_demo.ipynb](demo/inference_demo.ipynb).
A notebook demo can be found in [demo/inference_demo.ipynb](../demo/inference_demo.ipynb).
## Train a model
## Train a model
...
@@ -133,7 +133,7 @@ If you want to specify the working directory in the command, you can add an argu
...
@@ -133,7 +133,7 @@ If you want to specify the working directory in the command, you can add an argu
Optional arguments are:
Optional arguments are:
-`--validate` (**strongly recommended**): Perform evaluation at every k (default value is 1, which can be modified like [this](configs/mask_rcnn_r50_fpn_1x.py#L174)) epochs during the training.
-`--validate` (**strongly recommended**): Perform evaluation at every k (default value is 1, which can be modified like [this](../configs/mask_rcnn_r50_fpn_1x.py#L174)) epochs during the training.
-`--work_dir ${WORK_DIR}`: Override the working directory specified in the config file.
-`--work_dir ${WORK_DIR}`: Override the working directory specified in the config file.
-`--resume_from ${CHECKPOINT_FILE}`: Resume from a previous checkpoint file.
-`--resume_from ${CHECKPOINT_FILE}`: Resume from a previous checkpoint file.
...
@@ -155,7 +155,7 @@ Here is an example of using 16 GPUs to train Mask R-CNN on the dev partition.
...
@@ -155,7 +155,7 @@ Here is an example of using 16 GPUs to train Mask R-CNN on the dev partition.
./tools/slurm_train.sh dev mask_r50_1x configs/mask_rcnn_r50_fpn_1x.py /nfs/xxxx/mask_rcnn_r50_fpn_1x 16
./tools/slurm_train.sh dev mask_r50_1x configs/mask_rcnn_r50_fpn_1x.py /nfs/xxxx/mask_rcnn_r50_fpn_1x 16
```
```
You can check [slurm_train.sh](tools/slurm_train.sh) for full arguments and environment variables.
You can check [slurm_train.sh](../tools/slurm_train.sh) for full arguments and environment variables.
If you have just multiple machines connected with ethernet, you can refer to
If you have just multiple machines connected with ethernet, you can refer to
@@ -197,7 +197,7 @@ More models with different backbones will be added to the model zoo.
...
@@ -197,7 +197,7 @@ More models with different backbones will be added to the model zoo.
**Notes:**
**Notes:**
- Please refer to [Hybrid Task Cascade](configs/htc/README.md) for details and more a powerful model (50.7/43.9).
- Please refer to [Hybrid Task Cascade](../configs/htc/README.md) for details and more a powerful model (50.7/43.9).
### SSD
### SSD
...
@@ -214,54 +214,54 @@ More models with different backbones will be added to the model zoo.
...
@@ -214,54 +214,54 @@ More models with different backbones will be added to the model zoo.
### Group Normalization (GN)
### Group Normalization (GN)
Please refer to [Group Normalization](configs/gn/README.md) for details.
Please refer to [Group Normalization](../configs/gn/README.md) for details.
### Weight Standardization
### Weight Standardization
Please refer to [Weight Standardization](configs/gn+ws/README.md) for details.
Please refer to [Weight Standardization](../configs/gn+ws/README.md) for details.
### Deformable Convolution v2
### Deformable Convolution v2
Please refer to [Deformable Convolutional Networks](configs/dcn/README.md) for details.
Please refer to [Deformable Convolutional Networks](../configs/dcn/README.md) for details.
### Libra R-CNN
### Libra R-CNN
Please refer to [Libra R-CNN](configs/libra_rcnn/README.md) for details.
Please refer to [Libra R-CNN](../configs/libra_rcnn/README.md) for details.
### Guided Anchoring
### Guided Anchoring
Please refer to [Guided Anchoring](configs/guided_anchoring/README.md) for details.
Please refer to [Guided Anchoring](../configs/guided_anchoring/README.md) for details.
### FCOS
### FCOS
Please refer to [FCOS](configs/fcos/README.md) for details.
Please refer to [FCOS](../configs/fcos/README.md) for details.
### Grid R-CNN (plus)
### Grid R-CNN (plus)
Please refer to [Grid R-CNN](configs/grid_rcnn/README.md) for details.
Please refer to [Grid R-CNN](../configs/grid_rcnn/README.md) for details.
### GHM
### GHM
Please refer to [GHM](configs/ghm/README.md) for details.
Please refer to [GHM](../configs/ghm/README.md) for details.
### GCNet
### GCNet
Please refer to [GCNet](configs/gcnet/README.md) for details.
Please refer to [GCNet](../configs/gcnet/README.md) for details.
### HRNet
### HRNet
Please refer to [HRNet](configs/hrnet/README.md) for details.
Please refer to [HRNet](../configs/hrnet/README.md) for details.
### Mask Scoring R-CNN
### Mask Scoring R-CNN
Please refer to [Mask Scoring R-CNN](configs/ms_rcnn/README.md) for details.
Please refer to [Mask Scoring R-CNN](../configs/ms_rcnn/README.md) for details.
### Train from Scratch
### Train from Scratch
Please refer to [Rethinking ImageNet Pre-training](configs/scratch/README.md) for details.
Please refer to [Rethinking ImageNet Pre-training](../configs/scratch/README.md) for details.
### Other datasets
### Other datasets
We also benchmark some methods on [PASCAL VOC](configs/pascal_voc/README.md), [Cityscapes](configs/cityscapes/README.md) and [WIDER FACE](configs/wider_face/README.md).
We also benchmark some methods on [PASCAL VOC](../configs/pascal_voc/README.md), [Cityscapes](../configs/cityscapes/README.md) and [WIDER FACE](../configs/wider_face/README.md).
## Comparison with Detectron and maskrcnn-benchmark
## Comparison with Detectron and maskrcnn-benchmark