AaronJny / xyolo

一个高度封装的yolov3类库。
MIT License
35 stars 7 forks source link

Update yolo.py #2

Closed CofeCup closed 3 years ago

CofeCup commented 3 years ago

AaronJny 大大你好,opencv 读出来的颜色通道是按 BGR 的顺序,但 pillow 读取是按 RGB 的顺序,训练的时候 detect_image() 函数是使用 pillow 函数读取图片的,但用视频进行检测时是先用 opencv 读取的视频帧,所以这里需要加上将通道调换的代码,否则视频测试时输入 detect_and_draw_image() 的图片的通道顺序与训练时不一致。

AaronJny commented 3 years ago

收到,感谢,我晚上确认一下哈