Open hs7251912 opened 3 years ago
在yolov5_prune/modelsori.py下测试DarkNet, if name =='main': img_size = 608 cfg = 'cfg/yolov5s_v4_hand.cfg' arc = 'default' device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = Darknet(cfg, (img_size, img_size), arc=arc).to(device)
有警告: WARNING: smart bias initialization failure. WARNING: smart bias initialization failure. WARNING: smart bias initialization failure.
pytorch版本:1.8.1 python版本:3.7
代码使用torch版本为1.7,具体配置可参考yolov5第四版
在yolov5_prune/modelsori.py下测试DarkNet, if name =='main': img_size = 608 cfg = 'cfg/yolov5s_v4_hand.cfg' arc = 'default' device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
创建Darknet模型
有警告: WARNING: smart bias initialization failure. WARNING: smart bias initialization failure. WARNING: smart bias initialization failure.
pytorch版本:1.8.1 python版本:3.7