Skip to content

sacred config for faster rcnn #1358

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 25 commits into from
Jul 8, 2020
Merged

sacred config for faster rcnn #1358

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
25 commits
Select commit Hold shift + click to select a range
da5c626
move rcnn forward backward task to model zoo
Jerryzcn May 6, 2020
f71a984
revert https://github.com/dmlc/gluon-cv/pull/1249
Jerryzcn May 6, 2020
79ac948
fix
Jerryzcn May 6, 2020
f05fd36
fix
Jerryzcn May 6, 2020
3c2cac8
docstring
Jerryzcn May 6, 2020
d97c070
fix style
Jerryzcn May 6, 2020
8207c93
add docs
Jerryzcn May 6, 2020
51b3172
faster rcnn estimator
Jerryzcn Jun 9, 2020
adc3ad4
Merge remote-tracking branch 'upstream/master' into pr-rcnn
Jerryzcn Jun 9, 2020
5ba0fee
refactor
Jerryzcn Jun 10, 2020
c5ea693
move dataset to init
Jerryzcn Jun 16, 2020
dc25e53
lint
Jerryzcn Jun 16, 2020
3bc9ca2
Merge remote-tracking branch 'refs/remotes/upstream/master' into pr-rcnn
Jerryzcn Jun 16, 2020
72e0550
Merge remote-tracking branch 'upstream/estimator' into pr-rcnn
Jerryzcn Jun 16, 2020
e73985a
merge
Jerryzcn Jun 16, 2020
f3aaaba
disable sacred config for now
Jerryzcn Jun 17, 2020
988ce54
Merge branch 'estimator' of https://github.com/dmlc/gluon-cv into pr-…
Jerryzcn Jun 17, 2020
f56bd79
logger fix
Jerryzcn Jun 17, 2020
d46f892
fix fit
Jerryzcn Jun 17, 2020
304ce5b
autogluon integration
Jerryzcn Jun 30, 2020
df66311
Merge remote-tracking branch 'upstream/estimator' into pr-rcnn
Jerryzcn Jun 30, 2020
1c35190
fix small bug. training working
Jerryzcn Jun 30, 2020
c85b928
lint
Jerryzcn Jun 30, 2020
3bc987f
sacred config for faster rcnn
Jerryzcn Jul 8, 2020
b2cd33c
Merge remote-tracking branch 'upstream/estimator' into pr-rcnn
Jerryzcn Jul 8, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
lint
  • Loading branch information
@Jerryzcn
Jerryzcn committed Jun 30, 2020
commit c85b928605c9ef12d34ed47a7fd3c47fc2f9d897
4 changes: 2 additions & 2 deletions gluoncv/model_zoo/rcnn/rcnn.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -328,8 +328,8 @@ def custom_rcnn_fpn(pretrained_base=True, base_network_name='resnet18_v1b', norm
elif base_network_name == 'resnest50':
from ...model_zoo.resnest import resnest50
base_network = resnest50(pretrained=pretrained_base, dilated=False,
use_global_stats=use_global_stats, norm_layer=norm_layer,
norm_kwargs=norm_kwargs)
use_global_stats=use_global_stats, norm_layer=norm_layer,
norm_kwargs=norm_kwargs)
fpn_inputs_names = ['layers1_relu11_fwd', 'layers2_relu15_fwd', 'layers3_relu23_fwd',
'layers4_relu11_fwd']
elif base_network_name == 'resnest101':
Expand Down
5 changes: 2 additions & 3 deletions gluoncv/pipelines/estimators/rcnn/faster_rcnn.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -90,8 +90,8 @@ def _get_dataloader(net, train_dataset, val_dataset, train_transform, val_transf


def _get_testloader(net, test_dataset, num_devices, config):
"""Get faster rcnn test dataloader."""
if config.meta_arch == 'faster_rcnn':
"""Get faster rcnn dataloader."""
test_bfn = Tuple(*[Append() for _ in range(3)])
short = net.short[-1] if isinstance(net.short, (tuple, list)) else net.short
# validation use 1 sample per device
Expand Down Expand Up @@ -488,10 +488,9 @@ def load_parameters(self, parameters, multi_precision=False):
param_dict = self.net._collect_params_with_prefix()
kwargs = {'ctx': None} if mx.__version__[:3] == '1.4' else {'cast_dtype': multi_precision,
'ctx': None}
for k, v in param_dict.items():
for k, _ in param_dict.items():
param_dict[k]._load_init(parameters[k], **kwargs)

def get_parameters(self):
"""Return model parameters"""
return self.net._collect_params_with_prefix()