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

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
logger fix
  • Loading branch information
@Jerryzcn
Jerryzcn committed Jun 17, 2020
commit f56bd79ae1c145886fcbef621c1a29830b0a2506
10 changes: 5 additions & 5 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 @@ -115,7 +115,7 @@ def _get_dataset(dataset, args):
return train_dataset, val_dataset, val_metric


class FasterRCNNEstimator():
class FasterRCNNEstimator:
""" Estimator for Faster R-CNN.
TODO: use base estimators.
"""
Expand Down Expand Up @@ -236,18 +236,18 @@ def __init__(self, config, logger=None):
param.initialize()
self.net.collect_params().reset_ctx(self.ctx)
# set up logger
logger.setLevel(logging.INFO)
self._logger.setLevel(logging.INFO)
log_file_path = self._cfg.save_prefix + '_train.log'
log_dir = os.path.dirname(log_file_path)
if log_dir and not os.path.exists(log_dir):
os.makedirs(log_dir)
fh = logging.FileHandler(log_file_path)
logger.addHandler(fh)
self._logger.addHandler(fh)
if self._cfg.custom_model:
logger.info(
self._logger.info(
'Custom model enabled. Expert Only!! Currently non-FPN model is not supported!!'
' Default setting is for MS-COCO.')
logger.info(self._cfg)
self._logger.info(self._cfg)
self.rpn_cls_loss = gluon.loss.SigmoidBinaryCrossEntropyLoss(from_sigmoid=False)
self.rpn_box_loss = gluon.loss.HuberLoss(rho=self._cfg.rpn_smoothl1_rho) # == smoothl1
self.rcnn_cls_loss = gluon.loss.SoftmaxCrossEntropyLoss()
Expand Down