search

cuuupid / darkflow

Altered version of darkflow with some key improvements to support YOLO and LabelBox
GNU General Public License v3.0
0 stars 0 forks source link

Can't load weights #1

Open colorfuldarkgray opened 5 years ago

colorfuldarkgray commented 5 years ago

Tried: flow --model cfg/tiny-yolo.cfg --load bin/tiny-yolo.weights Got: Parsing ./cfg/tiny-yolo.cfg Parsing cfg/tiny-yolo.cfg Loading bin/tiny-yolo.weights ... Traceback (most recent call last): File "/home/ruy/.virtualenvs/cv/bin/flow", line 7, in exec(compile(f.read(), file, 'exec')) File "/home/ruy/darkflow/flow", line 4, in cliHandler(sys.argv) File "/home/ruy/darkflow/darkflow/cli.py", line 26, in cliHandler tfnet = TFNet(FLAGS) File "/home/ruy/darkflow/darkflow/net/build.py", line 58, in init darknet = Darknet(FLAGS) File "/home/ruy/darkflow/darkflow/dark/darknet.py", line 27, in init self.load_weights() File "/home/ruy/darkflow/darkflow/dark/darknet.py", line 83, in load_weights wgts_loader = loader.create_loader(args) File "/home/ruy/darkflow/darkflow/utils/loader.py", line 105, in create_loader return load_type(path, cfg) File "/home/ruy/darkflow/darkflow/utils/loader.py", line 19, in init self.load(args) File "/home/ruy/darkflow/darkflow/utils/loader.py", line 77, in load walker.offset, walker.size) AssertionError: expect 64701556 bytes, found 108399816

Also tried: flow --model cfg/yolov1.cfg --load bin/yolov1.weights And got: Parsing ./cfg/yolov1.cfg Parsing cfg/yolov1.cfg Loading bin/yolov1.weights ... Traceback (most recent call last): File "/home/ruy/.virtualenvs/cv/bin/flow", line 7, in exec(compile(f.read(), file, 'exec')) File "/home/ruy/darkflow/flow", line 4, in cliHandler(sys.argv) File "/home/ruy/darkflow/darkflow/cli.py", line 26, in cliHandler tfnet = TFNet(FLAGS) File "/home/ruy/darkflow/darkflow/net/build.py", line 58, in init darknet = Darknet(FLAGS) File "/home/ruy/darkflow/darkflow/dark/darknet.py", line 27, in init self.load_weights() File "/home/ruy/darkflow/darkflow/dark/darknet.py", line 83, in load_weights wgts_loader = loader.create_loader(args) File "/home/ruy/darkflow/darkflow/utils/loader.py", line 105, in create_loader return load_type(path, cfg) File "/home/ruy/darkflow/darkflow/utils/loader.py", line 19, in init self.load(args) File "/home/ruy/darkflow/darkflow/utils/loader.py", line 77, in load walker.offset, walker.size) AssertionError: expect 203934260 bytes, found 789312988

What is this expected size? I downloaded weight files from pjreddie web

Thank you for sharing

colorfuldarkgray commented 5 years ago

Also tried yolo-tiny.weights from your drive with tiny-yolo.cfg (renamed as yolo-tiny.weights) and got the same size error type: AssertionError: expect 64701556 bytes, found 180357512