search

bubbliiiing / yolov4-pytorch

这是一个YoloV4-pytorch的源码，可以用于训练自己的模型。
MIT License
2.08k stars 612 forks source link

yolotraining.py中参数问题 #340

Open ChenMaolong opened 1 year ago

ChenMaolong commented 1 year ago

大佬，我的数据集是单类别小目标肺结节病灶的，类别为1不是自然图像，尺寸608608， yolo_trainning.py中 self.obj_ratio = 5 (input_shape[0] * input_shape[1]) / (416 * 2) self.cls_ratio = 1 (num_classes / 80)这里我是不是应该把416改为608，还有80改为1的？还有其它要改的地方吗？

ChenMaolong commented 1 year ago

608的改，80是固定的数是吗

bubbliiiing commented 1 year ago

你不用改……

ChenMaolong commented 1 year ago

你不用改…… 好，就是inputshape416或者608都是这个对吧，谢谢