search

Stinky-Tofu / Stronger-yolo

🔥Improve yolo with latest paper
MIT License
3 stars 0 forks source link

An error occurred when I ran test.py #65

Open caoabc opened 5 years ago

caoabc commented 5 years ago

When I ran test.py using the weight file you provided, the following problems occurred
InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. lhs shape= [1,1,1024,18] rhs shape= [1,1,1024,75] [[Node: save/Assign_390 = Assign[T=DT_FLOAT, _class=["loc:@YoloV3/conv6/weights"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](YoloV3/conv6/weights, save/RestoreV2/_781)]] [[Node: save/RestoreV2/_82 = _SendT=DT_FLOAT, client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device_incarnation=1, tensor_name="edge_88_save/RestoreV2", _device="/job:localhost/replica:0/task:0/device:CPU:0"]] Is the weight file mismatched? I hope to get your answer. Thanks

Stinky-Tofu commented 5 years ago

不好意思，我已经停止维护这份代码了，如果按照readme来做，应该是没问题的

zhangyufei1995 commented 5 years ago

have you solved the erro?