Spaces:
Build error
Build error
napatswift
commited on
Commit
·
254a9f5
1
Parent(s):
6ad06e7
Update weights
Browse files- model/text-det/psenet.pth +2 -2
- model/text-det/psenet.py +3 -3
model/text-det/psenet.pth
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f60596dbe0b716a90a83cbd8b70a4761980bde06567acdd23c88fc4ca3b922e8
|
3 |
+
size 351856997
|
model/text-det/psenet.py
CHANGED
@@ -147,7 +147,7 @@ default_hooks = dict(
|
|
147 |
timer=dict(type='IterTimerHook'),
|
148 |
logger=dict(type='LoggerHook', interval=100),
|
149 |
param_scheduler=dict(type='ParamSchedulerHook'),
|
150 |
-
checkpoint=dict(type='CheckpointHook', interval=
|
151 |
sampler_seed=dict(type='DistSamplerSeedHook'),
|
152 |
sync_buffer=dict(type='SyncBuffersHook'),
|
153 |
visualization=dict(
|
@@ -158,7 +158,7 @@ default_hooks = dict(
|
|
158 |
draw_gt=False,
|
159 |
draw_pred=False))
|
160 |
log_level = 'INFO'
|
161 |
-
log_processor = dict(type='LogProcessor', window_size=
|
162 |
load_from = None
|
163 |
resume = True
|
164 |
val_evaluator = dict(type='HmeanIOUMetric')
|
@@ -171,7 +171,7 @@ visualizer = dict(
|
|
171 |
max_epochs = 200
|
172 |
optim_wrapper = dict(
|
173 |
type='OptimWrapper', optimizer=dict(type='Adam', lr=0.001))
|
174 |
-
train_cfg = dict(type='EpochBasedTrainLoop', max_epochs=
|
175 |
val_cfg = dict(type='ValLoop')
|
176 |
test_cfg = dict(type='TestLoop')
|
177 |
param_scheduler = [dict(type='PolyLR', power=0.9, end=200)]
|
|
|
147 |
timer=dict(type='IterTimerHook'),
|
148 |
logger=dict(type='LoggerHook', interval=100),
|
149 |
param_scheduler=dict(type='ParamSchedulerHook'),
|
150 |
+
checkpoint=dict(type='CheckpointHook', interval=5),
|
151 |
sampler_seed=dict(type='DistSamplerSeedHook'),
|
152 |
sync_buffer=dict(type='SyncBuffersHook'),
|
153 |
visualization=dict(
|
|
|
158 |
draw_gt=False,
|
159 |
draw_pred=False))
|
160 |
log_level = 'INFO'
|
161 |
+
log_processor = dict(type='LogProcessor', window_size=100, by_epoch=True)
|
162 |
load_from = None
|
163 |
resume = True
|
164 |
val_evaluator = dict(type='HmeanIOUMetric')
|
|
|
171 |
max_epochs = 200
|
172 |
optim_wrapper = dict(
|
173 |
type='OptimWrapper', optimizer=dict(type='Adam', lr=0.001))
|
174 |
+
train_cfg = dict(type='EpochBasedTrainLoop', max_epochs=50, val_interval=20)
|
175 |
val_cfg = dict(type='ValLoop')
|
176 |
test_cfg = dict(type='TestLoop')
|
177 |
param_scheduler = [dict(type='PolyLR', power=0.9, end=200)]
|