zhreshold / mxnet-yolo

YOLO: You only look once real-time object detector
MIT License
239 stars 88 forks source link

MXNet built from source problem #31

Open jacky4323 opened 6 years ago

jacky4323 commented 6 years ago

Hi, I use my own MXNet pakages that built from source,beacause I also has some custom operator. As you mentioned in your readme file ,some custom operators are not presented in official MXNet Can I add these additional custom operators to my own MXNet source? Which operators should I copy to my own MXNet pakages? Any help would be appericated! Thanks a lot!

Best Regards, PengWei

zhreshold commented 6 years ago

src/operator/contrib/yolo_output*

jacky4323 commented 6 years ago

Hi @zhreshold ,

I copy yolo_output* to my own mxnet and use cmake to build and I comment #import tools.find_mxnet in line2 demo.py then get below errors Could you please tell me how I can solve these problems?

File "demo.py", line 100, in ctx, args.nms_thresh, args.force_nms) File "demo.py", line 43, in get_detector data_shape, mean_pixels, ctx=ctx) File "/home/jacky4323/BMXNet_yolo/mxnet-yolo/detect/detector.py", line 34, in init load_symbol, args, auxs = mx.model.load_checkpoint(model_prefix, epoch) File "/home/jacky4323/BMXNet_yolo/mxnet/python/mxnet/model.py", line 420, in load_checkpoint symbol = sym.load('%s-symbol.json' % prefix) File "/home/jacky4323/BMXNet_yolo/mxnet/python/mxnet/symbol/symbol.py", line 2494, in load check_call(_LIB.MXSymbolCreateFromFile(c_str(fname), ctypes.byref(handle))) File "/home/jacky4323/BMXNet_yolo/mxnet/python/mxnet/base.py", line 146, in check_call raise MXNetError(py_str(_LIB.MXGetLastError())) mxnet.base.MXNetError: Failed loading Op stack_downsample of type stack_neighbor: [18:03:20] /home/jacky4323/BMXNet_yolo/mxnet/nnvm/src/core/op.cc:55: Check failed: op != nullptr Operator stack_neighbor is not registered

zhreshold commented 6 years ago

If you run the latest master, stack_neighbor operator shouldn't be required anymore.