hikapok / x-detector Goto Github PK
View Code? Open in Web Editor NEWLight-Head RCNN and One Novel Object Detector
License: Apache License 2.0
Light-Head RCNN and One Novel Object Detector
License: Apache License 2.0
I want to train/test xception model with light_head_rfcn_train.py, can you Provide a simple script?
thank you very much
Hi all,
I can't build libps_roi_align.so, running into following error:
X-Detector/cpp/PSROIPooling$ cmake .
X-Detector/cpp/PSROIPooling$ make
[ 10%] Building NVCC (Device) object CMakeFiles/cuda_compile_4.dir/cuda_compile_4_generated_rotated_ps_roi_align_grad_op.cu.o
sh: 1: cannot create -/rotated_ps_roi_align_grad_op.cpp1.ii: Directory nonexistent
CMake Error at cuda_compile_4_generated_rotated_ps_roi_align_grad_op.cu.o.cmake:219 (message):
Error generating
/data/open_source_projects/object_detection/X-Detector/cpp/PSROIPooling/CMakeFiles/cuda_compile_4.dir//./cuda_compile_4_generated_rotated_ps_roi_align_grad_op.cu.o
CMakeFiles/ps_roi_align.dir/build.make:84: recipe for target 'CMakeFiles/cuda_compile_4.dir/cuda_compile_4_generated_rotated_ps_roi_align_grad_op.cu.o' failed
make[2]: *** [CMakeFiles/cuda_compile_4.dir/cuda_compile_4_generated_rotated_ps_roi_align_grad_op.cu.o] Error 1
CMakeFiles/Makefile2:67: recipe for target 'CMakeFiles/ps_roi_align.dir/all' failed
make[1]: *** [CMakeFiles/ps_roi_align.dir/all] Error 2
Makefile:83: recipe for target 'all' failed
make: *** [all] Error 2
Env: OS Ubuntu 18, python 3.6.6, tensorflow-gpu==1.11.0, cuda-9.0
Any help is highly appreciated
Hi, thanks for sharing the code!
I went to cpp/PSROIPooling, and then executed cmake, and finally make, to generate the libps_roi_align.so file. But when I run the demo, I encountered this error:
tensorflow.python.framework.errors_impl.NotFoundError: ./cpp/PSROIPooling/libps_roi_align.so: undefined symbol: _ZN10tensorflow11GetNodeAttrERKNS_9AttrSliceENS_11StringPieceEPSs
I used exactly the same environment: tensorflow-1.6, python-3.5 and ubuntu 16.04. Also, I googled to find relevant topics but with no success.
Appreciate it if you can help.
Same error:
oi_align_op.h:25:61: fatal error: third_party/eigen3/unsupported/Eigen/CXX11/Tensor: No such file or directory
compilation terminated.
CMake Error at cuda_compile_4_generated_rotated_ps_roi_align_grad_op.cu.o.cmake:220 (message):
Error generating
/media/jintain/sg/ai/career/robot/ctirobot/detect/X-Detector/cpp/PSROIPooling/build/CMakeFiles/cuda_compile_4.dir//./cuda_compile_4_generated_rotated_ps_roi_align_grad_op.cu.o
CMakeFiles/ps_roi_align.dir/build.make:84: recipe for target 'CMakeFiles/cuda_compile_4.dir/cuda_compile_4_generated_rotated_ps_roi_align_grad_op.cu.o' failed
make[2]: *** [CMakeFiles/cuda_compile_4.dir/cuda_compile_4_generated_rotated_ps_roi_align_grad_op.cu.o] Error 1
CMakeFiles/Makefile2:72: recipe for target 'CMakeFiles/ps_roi_align.dir/all' failed
make[1]: *** [CMakeFiles/ps_roi_align.dir/all] Error 2
Makefile:83: recipe for target 'all' failed
make: *** [all] Error 2
Hi there,
I'm trying to run your light_head_simple_demo.py with provided model model.ckpt-121397
But while starting the script I run into out of memory issue: while restoring parameters from the checkpoint tf tries to occupy > 14Gb of GPU and falls because I have less than 16Gb.
But the model weight file has size less than 0,5Gb.
Does it really require >16Gb? Or I have somewhere wrong settings for the model? How much memory does it requires for that test?
For test I use tensorflow docker container with pre-installed Ubuntu16, cuda9.0, tensorflow1.6, python 3.5
Thanks
@HiKapok hi, have you ever try to train on COCO ? can the model be as good as the paper result (we can find its' mAP is close to yolov3, but 3 times fps to yolov2). Recently, I want to do human head detection, i want to use light-head-rcnn, can you give some advices? I am new in deep learning,thank you.
I have trained light_head_rcnn on VOC and i find the map is only about 20%.
i want to convert you light-head rcnn ckpt to pb,but i don't know the output_name of graph.
can you share the pbtxt for for your light-head-rcnn ckpt?
i want to test on image with checkpoint files you provide, can you provide a simple script for me. thank you very much!
I want to know what the backbone and the input size are on this map.
A declarative, efficient, and flexible JavaScript library for building user interfaces.
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google ❤️ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.