MatrixNetis a scale and aspect ratio aware deep learning architecture for object detection. We implemented matrixnets anchors (centers) and corners. For more details, please refer to the papers linked below.
We have two implementations based on Corners and Anchor (Centers):
Training and Evaluation Code
Code for reproducing the results in the following paper:
Selecting Layers in MatrixNets
One of the capabilities offered by MatrixNets is to be able to choose which layers to use for training and inference. Although we used 19 layers matrixnet in the paper, we implemented matrixnet here such that any matrixnet design can be specified by setting the
layer_range variable in the config file. The
layer_range is defined as a 3D matrix were the outer matrix is 5x5, and each entry of this matrix is either a 1D matrix of [y_min, y_max, x_min, x_max] or -1 if we do not want to include this layer.
In the paper, we use a 19-layer MatrixNet by ignoring the left top and bottom right corners of the 5x5 matrix. The range for the base layer (top left) is [24,48,24,48].
The corresonding layer range would look like:
[[[0,48,0,48],[48,96,0,48],[96,192,0,48], -1, -1],
[-1, -1, [0,192,384,2000],[192,384,384,2000],[384,2000,384,2000]]]
Note that we extended the range for the layers on the boundary to include any objects that are out of range.
Following table gives the AP for Corners and Anchors with different backbones (from the paper):
Note that these numbers are reported from the validation set, whereas the final numbers in the paper are reported on the test set.
Sample Images with Detections for both Architectures
Please first install Anaconda and create an Anaconda environment using the provided package list.
conda create --name matrixnets --file packagelist_conda.txt
After one creates the environment, activate it.
source activate matrixnets
Alternatively, one can use pip and install all packages from the requirements file. Note we are using python 3.6+. Torch 1.2.0 and torchvision 0.4.0
pip install -r requirements.txt
Our current implementation only supports GPU, so one needs a GPU and need to have CUDA(9+) installed on your machine.
cd <Matrixnet dir>/external make
Downloading MS COCO Data
- Download the training/validation split we use in our paper from here
- Unzip the file and place
- Download the images (2017 Train and 2017 Val) from here
- Create 2 directories,
- Copy the training/validation/testing images to the corresponding directories according to the annotation files
Training and Evaluation
To train and evaluate a network, one will need to create a configuration file, which defines the hyperparameters, and a model file, which defines the network architecture. The configuration file should be in JSON format and placed in
config/. Each configuration file should have a corresponding model file in
models/ (specified by
snapshot_name in the config file). i.e.
To train a model:
python train.py <config_file>
We provided four different configuration files under config directory for training both MatrixNetCorners and MatrixNetCenters.
To train MatrixNets:
python train.py MatrixNetsCornersResnet50
To evaluate the trained model:
python test.py <config_file> --testiter <iter> --split validation
We provide pre-trained models for Resnet-50 and Resnet-152 for both Anchors and Corners.
Please copy the pre-trained models into the following directory under matrixnets.
cache_name is the name of the directory specified in
name should be in the format
Note that the results might be slightly different from the paper (+/- 0.2 MAP) since we reproduced all experiments using only 4 GPUs. We could not fit the batch size of 23 for the anchors' experiments, so we ran the experiments for longer iterations to compensate for the smaller batch size.
List of avialble configuration options:
|dataset||Specify standard data||MSCOCO|
|batch_size||Specify batch size||At least 1|
|chunk_sizes||Size of chunk as a array of dim #GPU that sums to batch_size|
|model_name||specifying model (also picks the sampling function with the same name)||MatrixNetsCorners, MatrixNetAnchors|
|train_split||Spcify train set|
|val_split||Specify Validation Set|
|opt_algo||Specify Optimization Algorithm||adam|
|learning_rate||Specify learning rate|
|decay_rate||Specify learning rate decay|
|max_iter||Maximum number of Iterations|
|stepsize||Number of iterations for each learning rate decay|
|cache_dir||directory to store snapshots|
|data_dir||directory data is stored|
|rand_scale_min||Random Scaling Minimum Limt|
|rand_scale_max||Random Scaling Maximum Limt|
|rand_scale_step||Random Scaling Steps|
|input_size||Training image size||1d list that looks like [width, height]|
|output_kernel_size||This helps smoothing the heatmaps to get the max detections|
|base_layer_rangge||the input size for images in matrixnetanchors.||[y_min, y_max, x_min, x_max]|
|layers_range||3D matrix of Layer Range -1 inbdicating which layer to ignore|
|test_image_max_dim||max dim of input image|
|test_scales||test scales (if you want to test doing multiscale)||List of scales|
|test_flip_images||flip flag||True, False|
|cutout||cutout flag||True, False|
|top_k||Number of top k detections per layer||Integer|
|categories||number of classes|
|max_per_image||Max detections per image|
|merge_bbox||Merge bbox flag||True, False|
|weight_exp||exponential weighting specification|
|backbone||Backbone for Matrix Nets||resnet50, resnet100, resnet152, resnext101|
Subscribe to Python Awesome
Get the latest posts delivered right to your inbox