Comments (3)
Hi,
I changed the generator with the following function according to the tf2.4 documentation (https://www.tensorflow.org/api_docs/python/tf/keras/Model#fit) and it seems it works with batch size of 1. For batch size > 1 there is another issue according to #14
def faster_rcnn_generator(dataset, anchors, hyper_params):
"""Tensorflow data generator for fit method, yielding inputs and outputs.
inputs:
dataset = tf.data.Dataset, PaddedBatchDataset
anchors = (total_anchors, [y1, x1, y2, x2])
these values in normalized format between [0, 1]
hyper_params = dictionary
outputs:
yield inputs, outputs
"""
while True:
for image_data in dataset:
img, gt_boxes, gt_labels = image_data
bbox_deltas, bbox_labels = calculate_rpn_actual_outputs(anchors, gt_boxes, gt_labels, hyper_params)
yield (img, gt_boxes, gt_labels, bbox_deltas, bbox_labels),
from tf-faster-rcnn.
make virtual env with tensorflow version 2.1?
tf ver 2.4 makes errors like above.
from tf-faster-rcnn.
Hi @colindecourt I encounter a error while doing with tensorflow 2.1 any idea how to handle that problem? I already open an issue #21. Feel free to drop by.
from tf-faster-rcnn.
Related Issues (20)
- HOW to train my own dataset? HOT 1
- ValueError: Dimension 2 in both shapes must be equal, but are 4 and 1. Shapes are [?,1500,4] and [?,1500,1]. for '{{node roi_deltas/Select_1}} = Select[T=DT_FLOAT](roi_deltas/ExpandDims_9, roi_deltas/GatherV2_1, roi_deltas/zeros_like_1)' with input shapes: [?,1500,1], [?,1500,4], [?,1500,4].
- No such file or directory
- Training strategy HOT 1
- Erorr in reg_loss HOT 13
- 请问作者,voc2007和voc2012数据集训练时是怎么存放的?
- 怎么指定自己的VOC数据集
- Is there a trained checkpoint or weight h5 file?
- Question about epochs and learning rate selection
- Predict Error HOT 1
- Data Request
- variances in train_utils.py
- Bounding box resizing in preprocessing??
- Error occurred when finalizing GeneratorDataset iterator
- 数据集下载 HOT 3
- ValueError: The two structures don't have the same sequence length. Input structure has length 0, while shallow structure has length 9. HOT 3
- ValueError: The two structures don't have the same sequence length. Input structure has length 0, while shallow structure has length 9. HOT 1
- Does it support multi-GPU training? HOT 1
- ValueError: Dimension 2 in both shapes must be equal, but are 4 and 1. Shapes are [?,1500,4] and [?,1500,1]. for 'roi_deltas/Select' (op: 'Select') with input shapes: [?,1500,1], [?,1500,4], [?,1500,4]. HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from tf-faster-rcnn.