$ python demo.py [--demo /path/to/image/or/folder] [--show_mask]

xingyizhou / ExtremeNet

Bottom-up Object Detection by Grouping Extreme and Center Points

BSD 3-Clause "New" or "Revised" License

1.03k stars 174 forks source link

$ python demo.py [--demo /path/to/image/or/folder] [--show_mask] #12

Closed ghost closed 5 years ago

ghost commented 5 years ago

Hi,man! I should try demo.py in terminal,but code find bug! loading parameters: cache/ExtremeNet_250000.pkl building neural network... module_file: models.ExtremeNet total parameters: 198531504 loading parameters... loading from cache/ExtremeNet_250000.pkl Running Traceback (most recent call last): File "demo.py", line 150, in <module> height, width = image.shape[0:2] AttributeError: 'NoneType' object has no attribute 'shape' Please tell me where bug ? THK!

canghaiyunfan commented 5 years ago

@MingkunGao args.demo is None,you need to give the image path or folders of demo.