search

xueeinstein / darknet-vis

Visualize YOLO feature map in prediction for easily checking your model performance
Other
71 stars 8 forks source link

Usage #2

Open yoloboy2500 opened 5 years ago

yoloboy2500 commented 5 years ago

Hi,

To use this, do we need to completely install this version of darknet, or can we simply add a few of the classes included here?

Thanks.

xueeinstein commented 5 years ago

@yoloboy2500 There are two methods to use this version of darknet.

  1. CLI API. Build this version of darknet and add visualization=1 to net config, like https://github.com/xueeinstein/darknet-vis/blob/f53ddfc83ec2d747a272b6bd389da504b0e8b5f0/cfg/yolo.cfg#L25 Then you can use original darknet CLI API to detect your own image.
  2. Python API. Please see this example https://github.com/xueeinstein/darknet-vis/blob/f53ddfc83ec2d747a272b6bd389da504b0e8b5f0/python/darknet.py#L201