search

misads / easy_detection

一个简单方便的目标检测框架(PyTorch环境可直接运行，不需要cuda编译)，支持Faster_RCNN、Cascade_RCNN、Yolo系列、SSD等经典网络。
MIT License
256 stars 41 forks source link

Cascade_RCNN #19

Open bangbox2 opened 3 years ago

bangbox2 commented 3 years ago

Cascade_RCNN的model.py里面初始化没有config和kwargs参数

LiYuJieR commented 3 years ago

改成这样： class Model(BaseModel): def init(self, opt,**kwargs, logger=None): super(Model, self).init( kwargs,config=opt)

misads commented 3 years ago

最新版本已修复

TIMEXue commented 3 years ago

batchsize 只能为1吗

misads commented 2 years ago

batchsize 只能为1吗

已修复，现在可以为任意batch size