/
watchadoing
/
ArgicultureAI
Обзор
Документация
Войти
/
watchadoing
/
ArgicultureAI
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
Neural_Network/Point_fields_wheat/src/model.py
31 строка
996 B
Andrey Isaev
feat: add project to github
01 май 2025, 12:22
01 май 2025, 12:22
ff623b7
Код
Авторство
О чём код?
from tensorflow.keras.models import Model from tensorflow.keras.layers import Input, Conv2D, MaxPooling2D, UpSampling2D, concatenate # def unet_model(input_size=(640, 640, 1)): inputs = Input(input_size) # Encoder conv1 = Conv2D(64, (3, 3), activation='relu', padding='same')(inputs) pool1 = MaxPooling2D((2, 2))(conv1) conv2 = Conv2D(128, (3, 3), activation='relu', padding='same')(pool1) pool2 = MaxPooling2D((2, 2))(conv2) # Bottleneck conv3 = Conv2D(256, (3, 3), activation='relu', padding='same')(pool2) # Decoder up1 = UpSampling2D((2, 2))(conv3) concat1 = concatenate([up1, conv2], axis=-1) conv4 = Conv2D(128, (3, 3), activation='relu', padding='same')(concat1) up2 = UpSampling2D((2, 2))(conv4) concat2 = concatenate([up2, conv1], axis=-1) conv5 = Conv2D(64, (3, 3), activation='relu', padding='same')(concat2) # Output outputs = Conv2D(1, (1, 1), activation='sigmoid')(conv5) return Model(inputs, outputs)