- 04 4月, 2019 1 次提交
-
-
由 Zhang Liliang 提交于
Fix a bug. Romove the echo command in line 36: RUN conda install pytorch-nightly cudatoolkit=${CUDA} -c pytorch To enable conda installation of pytorch-nightly.
-
- 02 4月, 2019 1 次提交
-
-
由 Yihui_He 提交于
-
- 31 3月, 2019 1 次提交
-
-
由 Ouail 提交于
* add a FORCE_CUDA flag Following discussion [here](https://github.com/facebookresearch/maskrcnn-benchmark/issues/167), this seemed the best solution * Update Dockerfile * Update setup.py * add FORCE_CUDA as an ARG * modified: docker/Dockerfile modified: setup.py * small fix to readme of demo * remove test print * keep ARG_CUDA * remove env value and use the one from ARG * keep same formatting as source * change proposed by @miguelvr * Update INSTALL.md
-
- 27 3月, 2019 1 次提交
-
-
由 Francisco Massa 提交于
This reverts commit f0318794.
-
- 26 3月, 2019 3 次提交
-
-
由 Miguel Varela Ramos 提交于
* fixes to dockerfile * replaces local installation by git clone
-
由 Miguel Varela Ramos 提交于
* Merge branch 'master' of /home/braincreator/projects/maskrcnn-benchmark with conflicts. * rolls back the breaking AT dispatch changes (#555) * revert accidental docker changes * revert accidental docker changes (2)
-
由 kaiJIN 提交于
* support for any one cuda device * Revert "support for any one cuda device" This reverts commit 0197e4e2ef18ec41cc155f3ae2a0face5b77e1e9. * support runnning for anyone cuda device * using safe CUDAGuard rather than intrinsic CUDASetDevice * supplement a header dependency (test passed) * Support for arbitrary GPU device. * Support for arbitrary GPU device. * add docs for two method to control devices
-
- 25 3月, 2019 1 次提交
-
-
由 Bernhard Schäfer 提交于
-
- 13 3月, 2019 1 次提交
-
-
由 Csaba Botos 提交于
-
- 12 3月, 2019 2 次提交
-
-
由 Francisco Massa 提交于
-
由 Soumith Chintala 提交于
Fix dispatch breakage
-
- 11 3月, 2019 1 次提交
-
-
由 vishwakftw 提交于
-
- 10 3月, 2019 1 次提交
-
-
由 Bernhard Schäfer 提交于
-
- 08 3月, 2019 2 次提交
-
-
由 Stzpz 提交于
* Added a timer to benchmark model inference time in addition to total runtime. * Updated FBNet configs and included some baselines benchmark results. * Added a unit test for detectors. * Add links to the models
-
由 Francisco Massa 提交于
-
- 05 3月, 2019 2 次提交
-
-
由 Baptiste Metge 提交于
* fix INSTALL.md * fix PR * Update INSTALL.md
-
由 Bernhard Schäfer 提交于
-
- 01 3月, 2019 1 次提交
-
-
由 Erik 提交于
-
- 28 2月, 2019 2 次提交
-
-
由 Alexander Pacha 提交于
Using existing get_world_size to prevent AttributeError 'torch.distributed' has no attribute 'is_initialized'. (#511)
-
由 Cheng-Yang Fu 提交于
* Add new section "Projects using maskrcnn-benchmark". * Update README.md update the format. * Update README.md * Add coco_2017_train and coco_2017_val * Update README.md Add the instructions about COCO_2017 * Update the pip install. Adding tqdm which is used in engine/inference.py
-
- 26 2月, 2019 1 次提交
-
-
由 zimenglan 提交于
* make pixel indexes 0-based for bounding box in pascal voc dataset * replacing all instances of torch.distributed.deprecated with torch.distributed * replacing all instances of torch.distributed.deprecated with torch.distributed * add GroupNorm * add GroupNorm -- sort out yaml files * use torch.nn.GroupNorm instead, replace 'use_gn' with 'conv_block' and use 'BaseStem'&'Bottleneck' to simply codes * modification on 'group_norm' and 'conv_with_kaiming_uniform' function * modification on yaml files in configs/gn_baselines/ and reduce the amount of indentation and code duplication * use 'kaiming_uniform' to initialize resnet, disable gn after fc layer, and add dilation into ResNetHead * agnostic-regression for bbox * please set 'STRIDE_IN_1X1' to be 'False' when backbone use GN * add README.md for GN
-
- 23 2月, 2019 1 次提交
-
-
由 Preston Parry 提交于
There were two `RESNETS` sections, which overrode each other, leading to error messages like: ``` RuntimeError: Error(s) in loading state_dict for GeneralizedRCNN: size mismatch for backbone.fpn.fpn_inner1.weight: copying a param with shape torch.Size([256, 256, 1, 1]) from checkpoint, the shape in current model is torch.Size([1024, 256, 1, 1]). ... size mismatch for roi_heads.mask.feature_extractor.mask_fcn1.weight: copying a param with shape torch.Size([256, 256, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 1024, 3, 3]). ``` This just combines them back into a single section, while maintaining all param values. That got the model running again for me.
-
- 22 2月, 2019 1 次提交
-
-
由 Rene Bidart 提交于
-
- 20 2月, 2019 1 次提交
-
-
由 Stzpz 提交于
* Supported any feature map size for average pool. * Different models may have different feature map size. * Used registry to register keypoint and mask heads. * Passing in/out channels between modules when creating the model. Passing in/out channels between modules when creating the model. This simplifies the code to compute the input channels for feature extractors and makes the predictors independent of the backbone architectures. * Passed in_channels to rpn and head builders. * Set out_channels to model modules including backbone and feature extractors. * Moved cfg.MODEL.BACKBONE.OUT_CHANNELS to cfg.MODEL.RESNETS.BACKBONE_OUT_CHANNELS as it is not used by all architectures. Updated config files accordingly. For new architecture modules, the return module needs to contain a field called 'out_channels' to indicate the output channel size. * Added unit test for box_coder and nms. * Added FBNet architecture. * FBNet is a general architecture definition to support efficient architecture search and MaskRCNN2GO. * Included various efficient building blocks (inverted residual, shuffle, separate dw conv, dw upsampling etc.) * Supported building backbone, rpn, detection, keypoint and mask heads using efficient building blocks. * Architecture could be defined in `fbnet_modeldef.py` or in `cfg.MODEL.FBNET.ARCH_DEF` directly. * A few baseline architectures are included. * Added various unit tests. * build and run backbones. * build and run feature extractors. * build and run predictors. * Added a unit test to verify all config files are loadable.
-
- 19 2月, 2019 6 次提交
-
-
由 zimenglan 提交于
* make pixel indexes 0-based for bounding box in pascal voc dataset * replacing all instances of torch.distributed.deprecated with torch.distributed * replacing all instances of torch.distributed.deprecated with torch.distributed * add GroupNorm * add GroupNorm -- sort out yaml files * use torch.nn.GroupNorm instead, replace 'use_gn' with 'conv_block' and use 'BaseStem'&'Bottleneck' to simply codes * modification on 'group_norm' and 'conv_with_kaiming_uniform' function * modification on yaml files in configs/gn_baselines/ and reduce the amount of indentation and code duplication * use 'kaiming_uniform' to initialize resnet, disable gn after fc layer, and add dilation into ResNetHead * agnostic-regression for bbox * please set 'STRIDE_IN_1X1' to be 'False' when backbone use GN
-
由 xelmirage 提交于
I'm using pycharm to debug the code on a remote server, the remote debugging seems to be performed by pytest and it pops errors like: train_net.py E test setup failed file /tmp/pycharm_project_269/tools/train_net.py, line 79 def test(cfg, model, distributed): E fixture 'cfg' not found > available fixtures: cache, capfd, capfdbinary, caplog, capsys, capsysbinary, doctest_namespace, monkeypatch, pytestconfig, record_property, record_xml_attribute, recwarn, tmp_path, tmp_path_factory, tmpdir, tmpdir_factory > use 'pytest --fixtures [testpath]' for help on them. it seems the function name ‘test()’ has come conflict with pytest, so it may be better use another name.
-
由 Csaba Botos 提交于
-
由 Csaba Botos 提交于
* Remove Detectron dependency I have looked into the boxes.py to swap [these lines](https://github.com/facebookresearch/Detectron/blob/8170b25b425967f8f1c7d715bea3c5b8d9536cd8/detectron/utils/boxes.py#L51L52): ``` import detectron.utils.cython_bbox as cython_bbox import detectron.utils.cython_nms as cython_nms ``` ``` from maskrcnn_benchmark.structures.boxlist_ops import boxlist_iou from maskrcnn_benchmark.structures.boxlist_ops import boxlist_nms ``` However some functions are missing from the `boxlist_ops` like the [`soft_nms`](https://github.com/facebookresearch/Detectron/blob/master/detectron/utils/cython_nms.pyx#L98L203) . So I just tried to modify the `maskrcnn-benchmark/tools/cityscapes/convert_cityscapes_to_coco.py` script. Here we have `polys_to_boxes` function from `segms.py` and I could not find its analogous in the maskrcnn_benchmark lib. It seems to me that the original function in `segms.py` is using pure lists so I just wrote two auxiliary functions reusing the boxList's convert method( https://github.com/facebookresearch/maskrcnn-benchmark/blob/master/maskrcnn_benchmark/structures/bounding_box.py#L67L70 ) and Detectron's polys_to_boxes ( https://github.com/facebookresearch/Detectron/blob/b5dcc0fe1d091cb70f9243939258215dd63e3dfa/detectron/utils/segms.py#L135L140 ): ``` def poly_to_box(poly): """Convert a polygon into a tight bounding box.""" x0 = min(min(p[::2]) for p in poly) x1 = max(max(p[::2]) for p in poly) y0 = min(min(p[1::2]) for p in poly) y1 = max(max(p[1::2]) for p in poly) box_from_poly = [x0, y0, x1, y1] return box_from_poly def xyxy_to_xywh(xyxy_box): xmin, ymin, xmax, ymax = xyxy_box TO_REMOVE = 1 xywh_box = (xmin, ymin, xmax - xmin + TO_REMOVE, ymax - ymin + TO_REMOVE) return xywh_box ``` * removed leftovers * Update convert_cityscapes_to_coco.py
-
由 Preston Parry 提交于
Finishing the clean up in https://github.com/facebookresearch/maskrcnn-benchmark/pull/455, unsetting the proper variable. In general, thanks for making this so easy to install! I'd run into all kinds of versioning issues (version of Ubuntu not playing nicely with versions of CUDA/pytorch/libraries) trying to install other libraries implementing these algorithms. I'm super impressed by the quality of support, and the easy install, for this library.
-
由 Csaba Botos 提交于
the env variable is misused in the current version
-
- 18 2月, 2019 2 次提交
-
-
由 Preston Parry 提交于
The previous instruction examples assumed that the directory `~/github` existed, and did not include any check to create it if the directory did not exist. I updated to install in whatever directory the user is current in. I also updated to make it clear how the CUDA version is specified, and fixed a typo in activating the conda env.
-
由 Csaba Botos 提交于
A few addition: I added the top level directory `cityscapes` since the `tools/cityscapes/convert_cityscapes_to_coco.py` script has the directory structure `gtFine_trainvaltest/gtFine` hardcoded into it which is fine but was not clear at first. Also added a **Note** to warn people to install detectron as well, since the script uses `detectron.utils.boxes` and `detectron.utils.segm` modules which has further dependencies in the detectron lib.
-
- 15 2月, 2019 5 次提交
-
-
由 Ren Jin 提交于
-
由 Cheng-Yang Fu 提交于
* Add RetinetNet parameters in cfg. * hot fix. * Add the retinanet head module now. * Add the function to generate the anchors for RetinaNet. * Add the SigmoidFocalLoss cuda operator. * Fix the bug in the extra layers. * Change the normalizer for SigmoidFocalLoss * Support multiscale in training. * Add retinannet training script. * Add the inference part of RetinaNet. * Fix the bug when building the extra layers in retinanet. Update the matching part in retinanet_loss. * Add the first version of the inference of RetinaNet. Need to check it again to see if is there any room for speed improvement. * Remove the retinanet_R-50-FPN_2x.yaml first. * Optimize the retinanet postprocessing. * quick fix. * Add script for training RetinaNet with ResNet101 backbone. * Move cfg.RETINANET to cfg.MODEL.RETINANET * Remove the variables which are not used. * revert boxlist_ops. Generate Empty BoxLists instead of [] in retinanet_infer * Remove the not used commented lines. Add NUM_DETECTIONS_PER_IMAGE * remove the not used codes. * Move retinanet related files under Modeling/rpn/retinanet * Add retinanet_X_101_32x8d_FPN_1x.yaml script. This model is not fully validated. I only trained it around 5000 iterations and everything is fine. * set RETINANET.PRE_NMS_TOP_N as 0 in level5 (p7), because previous setting may generate zero detections and could cause the program break. This part is used in original Detectron setting. * Fix the rpn only bug when the training ends. * Minor improvements * Comments and add Python-only implementation * Bugfix and remove commented code * keep the generalized_rcnn same. Move the build_retinanet inside build_rpn. * Add USE_C5 in the MODEL.RETINANET * Add two configs using P5 to generate P6. * fix the bug when loading the Caffe2 ImageNet pretrained model. * Reduce the code depulication of RPN loss and RetinaNet loss. * Remove the comment which is not used. * Remove the hard coded number of classes. * share the foward part of rpn inference. * fix the bug in rpn inference. * Remove the conditional part in the inference. * Bug fix: add the utils file for permute and flatten of the box prediction layers. * Update the comment. * quick fix. Adding import cat. * quick fix: forget including import. * Adjust the normalization part according to Detectron's setting. * Use the bbox reg normalization term. * Clean the code according to recent review. * Using CUDA version for training now. And the python version for training on cpu. * rename the directory to retinanet. * Make the train and val datasets are consistent with mask r-cnn setting. * add comment.
-
由 Himanshu Pandey 提交于
-
由 CoinCheung 提交于
-
由 Levi Viana 提交于
* Adding support to Caffe2 ResNeXt-152-32x8d-FPN-IN5k backbone for Mask R-CNN * Clean up * Fixing path_catalogs.py * Back to old ROIAlign_cpu.cpp file
-
- 13 2月, 2019 2 次提交
-
-
由 Rodrigo Berriel 提交于
-
由 Francisco Massa 提交于
-
- 12 2月, 2019 1 次提交
-
-
由 Francisco Massa 提交于
* Add RPN config files * Add more RPN models
-