tommyanqi / yolov3-pytorch Goto Github PK

I try to use mobilenet2 to replace darknet in yolo3. But I don't get good results in coco.

When I train a pascal datasets model using mobilenet, I faced the problem that is "File "train.py", line 188, in _create_model
model.load_state_dict(torch.load(model_path)). FileNotFoundError: [Errno 2] No such file or directory: ' /model.pth'"
Does it mean need a pretrained model? I downloaded a mobilenetv2 pretrained model on Pytorch, and appointed this model to weightfile in yolo_train.yml and modified the model_path in train.py(177 line)
, but I ran into another problem that is about model dict key error, it same means model is different from net structure. Could you please tell me how to train about what model? Much appreciation!