yolodet-pytorch icon indicating copy to clipboard operation
yolodet-pytorch copied to clipboard

reproduce the YOLO series of papers in pytorch, including YOLOv4, PP-YOLO, YOLOv5,YOLOv3, etc.

Results 25 yolodet-pytorch issues
Sort by recently updated
recently updated
newest added

https://github.com/wuzhihao7788/yolodet-pytorch/blob/master/docs/MODEL_ZOO_cn.md 比如 网络 | 预训练数据集 | 输入尺寸 | epoch | 显卡类型 | 推理时间(fps) | cocotools APval | AP50 | 百度网盘 | 谷歌网盘 | 配置文件 | 日志 -- | -- |...

你好,我想问下在ppyolo中添加的iou-aware分支,在inference时,原论文中的iou confidence是通过α=0.4次方的形式与class的confidence的0.6次方相乘,得到inference时的confidence。 但是在代码中,yolodet/models/heads/ppyolo.py中的get_eval_bboxes函数: new_obj = torch.pow(obj, (1 - self.iou_aware_factor)) * torch.pow(ioua, self.iou_aware_factor) 实现论文的new_obj以后还进行了-log(1/new_obj-1)的操作。 我这边实现了iou-aware后假设原new_obj为0.9,但是经过-log(1/new_obj-1)后续操作得到的confidence数值是2.19,造成了识别的confidence的范围不在0到1以内。 想请教下为什么代码中这样实现,还是说在其他地方有将其重新约束到(0,1)的操作?

是否有onnx转换支持。

``` Traceback (most recent call last): File "tools/train.py", line 144, in main() File "tools/train.py", line 140, in main train_detector(model,datasets,cfg,validate=args.validate,timestamp=timestamp,meta=meta) File "/home/wenchi/projects/yolodet-pytorch-2/yolodet/apis/train.py", line 161, in train_detector runner.run(data_loaders, cfg.workflow, cfg.total_epochs) File "/home/wenchi/projects/yolodet-pytorch-2/yolodet/apis/runner.py",...

首先感谢作者的分享! 使用yolov5的小伙伴注意,里面有一些bug和一些不完善的地方,直接使用即使训练300轮的mAP依然非常底!!!比如config配置,model,loss单分类的计算,数据增强等。建议自己用yolo5源码实现,不然bug都不知道在哪。

Traceback (most recent call last): File "tools/train.py", line 144, in main() File "tools/train.py", line 69, in main cfg = Config.fromfile(args.config) File "/home/tongyu/develop-project/yolodet-pytorch-master/yolodet/utils/config.py", line 174, in fromfile cfg_dict, cfg_text = Config._file2dict(filename)...

你好 @wuzhihao7788 我在mac上训练报错, Traceback (most recent call last): File "tools/train.py", line 144, in main() File "tools/train.py", line 69, in main cfg = Config.fromfile(args.config) File "/Users/youyin/Documents/人工智能研究项目/keypoint/yolodet-pytorch/yolodet/utils/config.py", line 173, in fromfile cfg_dict,...

感谢您这么棒的工作,不过有个疑问,cfg中没看到yolov3的工作目录,故有此一问。