ABCAI93
ABCAI93
> 您好，请问一下如何在detector.cpp中获取到识别结果的类别信息呢？ 我是这么实现的： 1. 首先给parse_yolov5函数增加一个vector容器存放类别信息，如bool Detector::parse_yolov5(const Blob::Ptr &blob,int net_grid,float cof_threshold, vector& o_rect,vector& o_rect_cof, vector& classIds) 2. 接着在detector.cpp中的第57行后增加一行代码，即把类别序号push进类别的容器类，如classIds.push_back(idx-5) 3. 在detector.cpp的第132行增加一个vector classIds; 然后把137行的双引号改成classIds[final[i]] 4. 修改detector.h文件中第18行的std::string name成int class_id 5. 最后在main.cpp的43行后可以把类别序号打印出来了：std::cout
Hello, thanks for your sharing. But I encounted the same problem, have you solved it? Besides, the speed of my training seems to be a little slow, may I ask...
> > 请问一下，执行run_ssd_example.py脚本报这个错该如何解决？ Traceback (most recent call last): > > File "run_ssd_example.py", line 57, in > > boxes, labels, probs = predictor.predict(image, 10, 0.4) > > File "D:\Projects\DeepLearning\MobileNetV3-SSD-master\vision\ssd\predictor.py", line 37,...
> Hi > > Did the code restart or crash? typically it should restart. It keeps restarting and then keeps repeating this error. I want to know what causes this...
> Hi, > > does the code never work or does it happen only a few times? Unfortunately, the code never work
> Hi, I run the code on rtx3090 and got the same problem.  yeah, pip install git+https://github.com/jonbarron/robust_loss_pytorch it works!