ShaoqingRen / faster_rcnn

Faster R-CNN
Other
2.68k stars 1.22k forks source link

error when run faster rcnn in (stage one of fast rcnn ) win10, GPU 1080 TI, Matlab 2018a ; i RUN demo without error ; any solve ? #195

Open ggghh opened 6 years ago

ggghh commented 6 years ago

fast_rcnn startup done GPU 1: free memory 9681179443 Use GPU 1 Cleared 0 solvers and 1 stand-alone nets


stage one proposal


aver_boxes_num = 3100, select top 2000 aver_boxes_num = 3104, select top 2000


stage one fast rcnn


conf: batch_size: 128 bbox_thresh: 0.5000 bg_thresh_hi: 0.5000 bg_thresh_lo: 0.1000 fg_fraction: 0.2500 fg_thresh: 0.5000 image_means: [224×224×3 single] ims_per_batch: 2 max_size: 1000 rng_seed: 6 scales: 600 test_binary: 0 test_max_size: 1000 test_nms: 0.3000 test_scales: 600 use_flipped: 1 use_gpu: 1

opts: cache_name: 'faster_rcnn_VOC2007_ZF_top-1_nms0_7_top2000_stage1_fast_rcnn' conf: [1×1 struct] do_val: 1 imdb_train: {[1×1 struct]} imdb_val: [1×1 struct] net_file: 'E:\faster_rcnn-master\faster_rcnn-master\models\pre_trained_models\ZF\ZF.caffemodel' roidb_train: {[1×1 struct]} roidb_val: [1×1 struct] snapshot_interval: 10000 solver_def_file: 'E:\faster_rcnn-master\faster_rcnn-master\models\fast_rcnn_prototxts\ZF\solver_30k40k.prototxt' val_interval: 2000 val_iters: 500

Preparing training data...Done. Preparing validation data...Done. Error using caffe_ glog check error, please check log and clear mex

Error in caffe.Solver/step (line 56) caffe_('solver_step', self.hSolver_self, iters);

Error in fast_rcnn_train>check_gpu_memory (line 216) caffe_solver.step(1);

Error in fast_rcnn_train (line 89) check_gpu_memory(conf, caffe_solver, num_classes, opts.do_val);

Error in Faster_RCNN_Train.do_fast_rcnn_train (line 7) model_stage.output_model_file = fast_rcnn_train(conf, dataset.imdb_train, dataset.roidb_train, ...

Error in script_faster_rcnn_VOC2007_ZF (line 53) model.stage1_fast_rcnn = Faster_RCNN_Train.do_fast_rcnn_train(conf_fast_rcnn, dataset, model.stage1_fast_rcnn, opts.do_val);

ishanashastri commented 5 years ago

@ggghh Did you figure out how to solve this issue?

Thanks :))