我正在使用TensorFlow使用胶囊网络训练自己的数据集。在训练mnist数据集时,它包含函数mnist.train.next_batch(批量大小)。如何使用TensorFlow替换此功能以训练自己的数据集?
答案 0 :(得分:0)
函数样本<script>
$(document).ready(function(){
$("#submit").click(function(e){
e.preventDefault();
product_name = $("#product_name").val();
color = $("#colorWell").val();
var formData = new FormData();
$.each($("#product_image"), function (i, obj) {
$.each(obj.files, function (j, file) {
formData.append('product_image[' + i + ']', file);
});
});
formData.append('product_name', product_name);
formData.append('color', color);
$.ajax({
type:"POST",
data:formData,
processData: false,
contentType: false,
url:"<?php echo base_url(); ?>admin/products",
success:function(data){
alert(data);
}
});
});
});
</script>
中的数据来自经过改组的训练数据集,然后返回批次进行训练。
您可以编写自己的batch_size
方法来执行相同的操作,也可以根据需要对其进行修改。然后在训练模型时类似地使用它。