使用mobilenet训练自己的数据

使用mobilenet训练自己的数据

实现背景:keras+ tensorflow

一、数据预处理文件:car2626data.py

#coding:utf-8  
import os  
import numpy as np  
import cv2
import pdb  
def load_data():  
    x_train = np.empty((263171,3,224,224),dtype="float32")  
    y_train = np.empty((263171,),dtype="uint8")
    x_test = np.empty((9940,3,224,224),dtype="float32") 
    y_test = np.empty((9940,),dtype="uint8")
    i = 0
    j = 0
    for line in open('/home/zhanghh/train_label_year.txt'):
        ss = line.split(' ')
        ori_img = "/home/zhanghh/data/" + ss[0]
#        pdb.set_trace()
        img1 = cv2.imread(ori_img)
        img2 = cv2.resize(img1,(224,224),interpolation=cv2.INTER_CUBIC)
        arr = np.asarray(img2,dtype="float32")
        x_train[i,:,:,:] = [arr[:,:,0],arr[:,:,1],arr[:,:,2]]
#        pdb.set_trace()
        y_train[i] = int(ss[-1])
        i+=1
    y_train = np.reshape(y_train,(len(y_train),1))
    for line in open('/home/zhanghh/test_label_year.txt'):
        nn = line.split(' ')
        or_img = "/home/zha
  • 4
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 7
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值