search

bubbliiiing / mask-rcnn-tf2

这是一个mask-rcnn-tf2的库，可以用于训练自己的模型。
MIT License
64 stars 9 forks source link

用shape数据集训练出错 #5

Open oneline-wsq opened 2 years ago

oneline-wsq commented 2 years ago

用shape数据集训练后报错，环境是按照文件中的要求配置的 Snipaste_2022-06-28_20-45-09 。

bubbliiiing commented 2 years ago

什么版本。 以及操作步骤。

oneline-wsq commented 2 years ago

scipy 1.4.1； numpy 1.18.4； matplotlib 3.2.1； opencv-python 4.2.0.34； tensorflow-gpu 2.2.0； tqdm 4.46.1； Pillow 8.2.0； h5py 2.10.0； scikit-image 0.16.2； pycocotools 2.0.4；

CUDA Version: 11.5

将shapes数据集放在datasets/before下，运行coco_ananotation,py, 再运行train.py， 就会报图片上的错误。

bubbliiiing commented 2 years ago

CUDA Version: 11.5，这个是？

oneline-wsq commented 2 years ago

CUDA的版本。 问题没有解决，但是已经换成别的网络了，谢谢大佬的耐心解答🥰

bubbliiiing commented 2 years ago

好吧，0 0