search

DetectionTeamUCAS / R2CNN_Faster-RCNN_Tensorflow

Rotational region detection based on Faster-RCNN.
MIT License
589 stars 186 forks source link

预测出错 #58

Closed SuHaoXD closed 5 years ago

SuHaoXD commented 5 years ago

用你训练好的的DOTA模型测试效果为什么是这样子？ 是需要修改哪里的cfg文件吗？ P1385_r P1385_h

SuHaoXD commented 5 years ago

使用的是DOTA cfg, demo_rh.py预测，顺便想问一下预测为什么生成两张图片？一张r 一张h

Smoothing97 commented 5 years ago

@SuHaoXD 同样的问题，我用了demo_rh.py，输入的图片是DOTA数据集里的val数据集，用val_crop.py切割成800*800，跑demo_rh.py也出现这样的情况。唯一一次出现大量正确标注的图片结果好像是因为我在demo_rh.py后又对这些图片结果跑了eval.py，绘制上去的box是gtbox....正在查到底哪里有问题。 PS：按照readme里写的应该是跑python inference.py --data_dir='/PATH/TO/DOTA/IMAGES_CROP/' --gpu='0'来对crop好的图片做检测，但是我放置好weight文件跑这个还是出现你图中的结果，不知道为什么。如果你有什么建议或者看法，期待和我交流：）

yangxue0827 commented 5 years ago

给一下你们checkpoint文件的绝对路径 @SuHaoXD @Smoothing97 可能是你们路径放的不对

yangxue0827 commented 5 years ago

你们这个情况像是加载了pretrain weights

SuHaoXD commented 5 years ago

你们这个情况像是加载了pretrain weights

已解决，原因是模型未能正确加载。DOTA权重文件应该放在一个文件夹中，然后再将这个文件夹放在trained_weights中。 @Smoothing97 感谢作者分享！

Smoothing97 commented 5 years ago

@SuHaoXD 我检查过这个问题，但是不太行，，我重新放置了一下，现在正在测试，过会儿回来更新：） 1、please download resnet50_v1、resnet101_v1 pre-trained models on Imagenet, put it to data/pretrained_weights. image 2、please download mobilenet_v2 pre-trained model on Imagenet, put it to data/pretrained_weights/mobilenet. 3、please download trained model by this project, put it to output/trained_weights. image image

Smoothing97 commented 5 years ago

更新回来了：我检查了一下cfgs.py里的VERSION = 'FasterRCNN_20180515_DOTA_v3'被我魔改成每天的日期了，可能找不到模型文件：）所以代码还是要好好看，随便改个什么就出BUG,233333