阴少辉(Shaohui Yin)
阴少辉(Shaohui Yin)
Hello. Have you solved the error?
The training, testing, validation file are located in /public/waymo/kitti_format/ The training, testing, validation file contain the files listed below: calib image_1 image_3 label_0 label_2 label_4 pose image_0 image_2 image_4 label_1...
checkpoint模型转为saved model是在tensorflow1.14版本转的,saved model转onnx是在tensorflow 2.2版本转的
若是用tensorflow 1.14转saved model为onnx,则内存快速被占满,之后被杀死
另外,我的输入图片是3072*2048的,显卡是RTX2060 6G,用ResNet_50计算,显存不够了。除了更换网络之外,能不能给一些建议?
在data目录下新建一个tfrecord文件夹解决了
https://github.com/yangxue0827/FPN_Tensorflow/issues/35#issuecomment-414103141 按照2楼的方法,先跑一下data/io/convert_data_to_tfrecord.py,在data/tfrecord下面会生成一个比数据集大很多的文件,然后跑train.py就行了。 这里的data/tfrecord文件夹需要新建,源代码data下面没有