這篇文章給大家分享的是有關(guān)Django中如何使用模型對(duì)上傳圖片預(yù)測(cè)的內(nèi)容。小編覺得挺實(shí)用的,因此分享給大家做個(gè)參考,一起跟隨小編過來看看吧。
目前成都創(chuàng)新互聯(lián)已為1000+的企業(yè)提供了網(wǎng)站建設(shè)、域名、網(wǎng)站空間、網(wǎng)站運(yùn)營(yíng)、企業(yè)網(wǎng)站設(shè)計(jì)、墨江網(wǎng)站維護(hù)等服務(wù),公司將堅(jiān)持客戶導(dǎo)向、應(yīng)用為本的策略,正道將秉承"和諧、參與、激情"的文化,與客戶和合作伙伴齊心協(xié)力一起成長(zhǎng),共同發(fā)展。1 預(yù)處理
(1)對(duì)上傳的圖片進(jìn)行預(yù)處理成100*100大小
def prepicture(picname): img = Image.open('./media/pic/' + picname) new_img = img.resize((100, 100), Image.BILINEAR) new_img.save(os.path.join('./media/pic/', os.path.basename(picname)))
(2)將圖片轉(zhuǎn)化成數(shù)組
def read_image2(filename): img = Image.open('./media/pic/'+filename).convert('RGB') return np.array(img)
2 利用模型進(jìn)行預(yù)測(cè)
def testcat(picname): # 預(yù)處理圖片 變成100 x 100 prepicture(picname) x_test = [] x_test.append(read_image2(picname)) x_test = np.array(x_test) x_test = x_test.astype('float32') x_test /= 255 keras.backend.clear_session() #清理session反復(fù)識(shí)別注意 model = Sequential() model.add(Conv2D(32, (3, 3), activation='relu', input_shape=(100, 100, 3))) model.add(Conv2D(32, (3, 3), activation='relu')) model.add(MaxPooling2D(pool_size=(2, 2))) model.add(Dropout(0.25)) model.add(Conv2D(64, (3, 3), activation='relu')) model.add(Conv2D(64, (3, 3), activation='relu')) model.add(MaxPooling2D(pool_size=(2, 2))) model.add(Dropout(0.25)) model.add(Flatten()) model.add(Dense(256, activation='relu')) model.add(Dropout(0.5)) model.add(Dense(4, activation='softmax')) sgd = SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True) model.compile(loss='categorical_crossentropy', optimizer=sgd, metrics=['accuracy']) model.load_weights('./cat/cat_weights.h6') classes = model.predict_classes(x_test)[0] # target = ['布偶貓', '孟買貓', '暹羅貓', '英國(guó)短毛貓'] # print(target[classes]) return classes
3 與Django結(jié)合
在views中調(diào)用模型進(jìn)行圖片分類
def catinfo(request): if request.method == "POST": f1 = request.FILES['pic1'] # 用于識(shí)別 fname = '%s/pic/%s' % (settings.MEDIA_ROOT, f1.name) with open(fname, 'wb') as pic: for c in f1.chunks(): pic.write(c) # 用于顯示 fname1 = './static/img/%s' % f1.name with open(fname1, 'wb') as pic: for c in f1.chunks(): pic.write(c) num = testcat(f1.name) # 有的數(shù)據(jù)庫(kù)id從1開始這樣就會(huì)報(bào)錯(cuò) # 因此原本數(shù)據(jù)庫(kù)中的id=0被系統(tǒng)改為id=4 # 遇到這樣的問題就加上 # if(num == 0): # num = 4 # 通過id獲取貓的信息 name = models.Catinfo.objects.get(id = num) return render(request, 'info.html', {'nameinfo': name.nameinfo, 'feature': name.feature, 'livemethod': name.livemethod, 'feednn': name.feednn, 'feedmethod': name.feedmethod, 'picname': f1.name}) else: return HttpResponse("上傳失??!")
感謝各位的閱讀!關(guān)于“Django中如何使用模型對(duì)上傳圖片預(yù)測(cè)”這篇文章就分享到這里了,希望以上內(nèi)容可以對(duì)大家有一定的幫助,讓大家可以學(xué)到更多知識(shí),如果覺得文章不錯(cuò),可以把它分享出去讓更多的人看到吧!
另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)scvps.cn,海內(nèi)外云服務(wù)器15元起步,三天無理由+7*72小時(shí)售后在線,公司持有idc許可證,提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國(guó)服務(wù)器、虛擬主機(jī)、免備案服務(wù)器”等云主機(jī)租用服務(wù)以及企業(yè)上云的綜合解決方案,具有“安全穩(wěn)定、簡(jiǎn)單易用、服務(wù)可用性高、性價(jià)比高”等特點(diǎn)與優(yōu)勢(shì),專為企業(yè)上云打造定制,能夠滿足用戶豐富、多元化的應(yīng)用場(chǎng)景需求。