Cyril666 commited on
Commit
ca7b6f4
·
1 Parent(s): 8e2468e

First model version

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,9 +13,9 @@ from maskrcnn_benchmark.config import cfg
13
 
14
 
15
  def infer(filepath):
16
- cfg.merge_from_file('configs/ctw/r50_baseline.yaml')
17
  # manual override some options
18
- cfg.merge_from_list(["MODEL.DEVICE", "cpu"])
19
 
20
  text_demo = TextDemo(
21
  cfg,
 
13
 
14
 
15
  def infer(filepath):
16
+ cfg.merge_from_file('./configs/ctw/r50_baseline.yaml')
17
  # manual override some options
18
+ #cfg.merge_from_list(["MODEL.DEVICE", "cpu"])
19
 
20
  text_demo = TextDemo(
21
  cfg,