search

misads / easy_detection

一个简单方便的目标检测框架(PyTorch环境可直接运行，不需要cuda编译)，支持Faster_RCNN、Cascade_RCNN、Yolo系列、SSD等经典网络。
MIT License
261 stars 41 forks source link

关于预测可视化的问题 #23

Closed TIMEXue closed 3 years ago

TIMEXue commented 3 years ago

get_started.md中原话： “4. 如果需要使用Tensorboard可视化预测结果，可以在上面的命令最后加上--vis参数。然后运行tensorboard --logdir results/cache查看检测的可视化结果”， 但是查看了代码发现，并没有预测结果可视化功能，请问是还没更新吗

misads commented 3 years ago

有的，在network/base_model.py里 https://github.com/misads/easy_detection/blob/7eee2bfbfd6f4fba505c3eb04e29efb049c86fbd/network/base_model.py#L93-L102

tensorboard需要根据不同的模型指定正确的目录，比如Faster RCNN，先运行这个

python3 eval.py --config configs/faster_rcnn_voc.yml --load pretrained/0_voc_FasterRCNN.pt --vis

再运行tensorboard

tensorboard --logdir results/faster_rcnn_voc