mask-rcnn-tf2
mask-rcnn-tf2 copied to clipboard
bubbliiiing
这是一个mask-rcnn-tf2的库，可以用于训练自己的模型。
#------------------------------------------------------------------# #获得RPN网络的预测结果，进行格式调整，把五个特征层的结果进行堆叠 #------------------------------------------------------------------# rpn_class_logits, rpn_class, rpn_bbox = [],[],[] for p in rpn_feature_maps: logits,classes,bbox = rpn([p]) rpn_class_logits.append(logits) rpn_class.append(classes) rpn_bbox.append(bbox) #------------------------------------------------------------------# #此时获得的rpn_class_logits、rpn_class、rpn_bbox的维度是 #rpn_class_logits : Batch_size, num_anchors, 2 #rpn_class : Batch_size, num_anchors, 2...
但是训练train的时候报错 Traceback (most recent call last): File "F:/pycharmprogram/mask-rcnn-tf2/train.py", line 296, in callbacks = callbacks File "C:\Users\12396\AppData\Roaming\Python\Python37\site-packages\tensorflow\python\keras\engine\training.py", line 66, in _method_wrapper return method(self, *args, **kwargs) File "C:\Users\12396\AppData\Roaming\Python\Python37\site-packages\tensorflow\python\keras\engine\training.py", line 815, in fit...
博主你好，我的运行环境是windows，显卡：RTX A6000 , CUDA 11.8，cuDNN 8.7.0, tensorflow-gpu 2.6.0，batch size为1。在训练到epoch2结束后程序运行结束，以下是报错的代码： 2023-03-14 16:05:09.195263: E tensorflow/stream_executor/cuda/cuda_event.cc:29] Error polling for event status: failed to query event: CUDA_ERROR_ILLEGAL_ADDRESS: an illegal memory access was encountered 2023-03-14...
导师你好， 我现在用mask-rcnn做训练的时候做了四个分类（2个不同种类的建筑和2个不同种类的道路）， 在可视化中想单独提取建筑类的结果。 尝试了直接减掉class_ids和用list的方式返回了class_names中我想要的分类，但是都失败了。 想问一下应该更改哪一部分可以做到单独提取我想要的分类？ 谢谢T^T
博主您好，我使用的版本是 cpu，tf2.2.0，使用shape数据集未更改其他参数，在运行train文件的epoch1时有如下报错，看了以往的问题与解决办法，尝试过换tf2.2-gpu或tf2.6-gpu，但均出现如图错误。请问应该如何修改？感谢您的回复！    
博主，您好，运行train.py时，在第一个epoch会报错： Epoch 00001: LearningRateScheduler reducing learning rate to 3e-06. Epoch 1/100 2022-07-18 09:42:46.656945: W tensorflow/core/grappler/optimizers/loop_optimizer.cc:906] Skipping loop optimization for Merge node with control input: mask_rcnn/proposal_targets/roi_assertion/AssertGuard/branch_executed/_8 Traceback (most recent call last):...
博主您好，shape数据集未更改其他参数，在运行train文件的epoch1时有如下报错，请问是图像尺寸的问题吗？如果是应该如何修改？感谢您的回复！ tensorflow.python.framework.errors_impl.InvalidArgumentError: indices[1] = 127306 is not in [0, 65472) [[{{node ROI/GatherV2_4}}]]
