search

misads / easy_detection

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

运行train.py时候出现UserWarning #30

Open YipChakwai opened 2 years ago

YipChakwai commented 2 years ago

运行指令时候python3 train.py --config configs/faster_rcnn_voc.yml，出现了如下警告： .\easy_detection-master\network\SSD\utils\nms.py:15: UserWarning: No NMS is available. Please upgrade torchvision to 0.3.0+ or compile c++ NMS using cd ext & python build.py build_ext develop warnings.warn('No NMS is available. Please upgrade torchvision to 0.3.0+ or compile c++ NMS ' 求大神解答！

misads commented 2 years ago

环境的问题吧，升级一下torchvision版本。