Ускорить расчет

@MConst · Регистрация: 16.04.2022

Студворк — интернет-сервис помощи студентам

Здравствуйте!

Есть цветное изображение с размером n * m. Есть заданные размеры фрагмента new_height и new_width. Нужно получить массив из всех фрагментов из полученного изображения.

Есть такой код

Кликните здесь для просмотра всего текста

Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
img_origin = plt.imread(r'...')
 
new_height, new_width = 300, 400
new_stride = 1
isExists = False
 
for i in range(0 - new_stride, img_origin.shape[0] - new_height, new_stride):
    for j in range(0 - new_stride, img_origin.shape[1] - new_width, new_stride):        
        temp = img_origin[i + 1: i + new_height + 1, j + 1: j + new_width + 1, :]
        temp = np.expand_dims(temp , axis = 0)
        
        if (isExists == False):
            dataSet = temp
            isExists = True
        else:
            dataSet = np.concatenate((dataSet, temp), axis= 0)
        
print(dataSet.shape)

Алгоритм работает, но долго.

Скажите, как можно написать так, чтобы работало намного быстрее?

Добавлено через 1 час 46 минут
Поправил код, так как в предыдущем варианте не корректно при new_stride > 1.

Кликните здесь для просмотра всего текста

Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
img_origin = plt.imread(r'...')
 
new_height, new_width = 300, 400
new_stride = 1
isExists = False
 
for i in range(0 - new_stride, img_origin.shape[0] - new_height, new_stride):   
    for j in range(0 - new_stride, img_origin.shape[1] - new_width, new_stride):       
        temp = img_origin[i + new_stride: i + new_height + new_stride, j + new_stride: j + new_width + new_stride, :]
        temp = np.expand_dims(temp, axis = 0)
        if (temp.shape[1] == new_height and temp.shape[2] == new_width):        
            if (isExists == False):
                dataSet = temp
                isExists = True
            else:
                dataSet = np.concatenate((dataSet, temp), axis= 0)
        
print(dataSet.shape)

@u235 · 13.05.2023, 20:05

В numpy есть функция numpy.lib.stride_tricks.sliding_window_v iew. Попробуйте использовать ее.

@MConst · 13.05.2023, 21:10 **[ТС]**

Пишу так

Python
1
np.lib.stride_tricks.sliding_window_view(img_origin, window_shape = (new_height, new_width, 3), axis = (0,1,2))

что-то не то. Или неправильно параметры задаю.

Добавлено через 1 минуту
И не понятно куда девать new_stride.

@u235 · 13.05.2023, 23:10

MConst,

Python
1
2
3
4
5
6
 import numpy as np
new_height, new_width = 3, 4
new_stride = 1
img_origin=np.random.randint(0,255,(3,8, 10), np.uint8)
res=np.lib.stride_tricks.sliding_window_view(img_origin, window_shape = (3,new_height, new_width))
res=res[0,::new_stride,::new_stride,:,:,:]

new_stride в сечении можно использовать

@MConst · 14.05.2023, 11:41 **[ТС]**

написал так

Python
1
np.lib.stride_tricks.sliding_window_view(img_origin, window_shape = (new_height, new_width, 3))[0, ::new_stride, ::new_stride, :, :, :].shape

Выдает такой размер (68, 1, 288, 384, 3).

Во-первых, изначально мой скрипт выдавал 408 экземпляров, то есть размер был (408, 288, 384, 3), а этот выдает 68. Почему то меньше.

Во-вторых, вторая ось не нужна. Хотя в конце можно использовать функцию np.squeeze. Не знаю можно ли сразу получить без второй оси.

@u235 · 14.05.2023, 16:45

MConst, у вас у картинки исходной какой shape?

@MConst · 14.05.2023, 19:33 **[ТС]**

(311, 653, 3)

@u235 · 14.05.2023, 20:47

MConst, для stride==1 результаты с вашим скриптом совпадают, только сильно быстрее все считается.

Python
1
2
3
4
5
6
import numpy as np
new_height, new_width = 300, 400
new_stride = 1
img_origin=np.random.randint(0,255,(311, 653, 3), np.uint8)
res=np.lib.stride_tricks.sliding_window_view(img_origin, window_shape = (new_height, new_width,3))[::new_stride ,::new_stride ,:,:,:,:]
res2=res.reshape(-1, new_height, new_width,3)

@MConst · 14.05.2023, 22:34 **[ТС]**

Супер! Спасибо!

Только вот не понятно, почему эта функция еще одну ось добавляет.

@u235 · 14.05.2023, 22:40

Сообщение от MConst

почему эта функция еще одну ось добавляет.

Потому что у вас картинка 311, 653, 3, по каждой оси есть движение окна, но т.к. window_shape у вас new_height, new_width,3 , то по последней оси формально движение есть, но оно нулевое. Это и дает еще одну ось.

@MConst · 14.05.2023, 22:42 **[ТС]**

Понятно

Новые блоги и статьи Все статьи Все блоги /
Автозаполнение реквизита при выборе элемента справочника Maks 27.03.2026 Программный код из решения ниже на примере нетипового документа "ЗаявкаНаРемонтСпецтехники" разработанного в конфигурации КА2. При выборе "Спецтехники" (Тип Справочник. Спецтехника), заполняется. . .	Сумматор с применением элементов трёх состояний. Hrethgir 26.03.2026 Тут. https:/ / fips. ru/ EGD/ ab3c85c8-836d-4866-871b-c2f0c5d77fbc Первый документ красиво выглядит, но без схемы. Это конечно не даёт никаких плюсов автору, но тем не менее. . . всё может быть. . .	Автозаполнение реквизитов при создании документа Maks 26.03.2026 Программный код из решения ниже размещается в модуле объекта документа, в процедуре "ПриСозданииНаСервере". Алгоритм проверки заполнения реализован для исключения перезаписи значения реквизита,. . .	Команды формы и диалоговое окно Maks 26.03.2026 1. Команда формы "ЗаполнитьЗапчасти". Программный код из решения ниже на примере нетипового документа "ЗаявкаНаРемонтСпецтехники" разработанного в конфигурации КА2. В качестве источника данных. . .
Кому нужен AOT? DevAlt 26.03.2026 Решил сделать простой ланчер Написал заготовку: dotnet new console --aot -o UrlHandler var items = args. Split(":"); var tag = items; var id = items; var executable = args;. . .	Отправка уведомления на почту при изменении наименования справочника Maks 24.03.2026 Программная отправка письма электронной почты на примере изменения наименования типового справочника "Склады" в конфигурации БП3. Перед реализацией необходимо выполнить настройку системной учетной. . .	модель ЗдравоСохранения 5. Меньше увольнений- больше дохода! anaschu 24.03.2026 Теперь система здравосохранения уменьшает количество увольнений. 9TO2GP2bpX4 a42b81fb172ffc12ca589c7898261ccb/ https:/ / rutube. ru/ video/ a42b81fb172ffc12ca589c7898261ccb/ Слева синяя линия -. . .	Midnight Chicago Blues kumehtar 24.03.2026 Такой Midnight Chicago Blues, знаешь?. . Когда вечерние улицы становятся ночными, а ты не можешь уснуть. Ты идёшь в любимый старый бар, и бармен наливает тебе виски. Ты смотришь на пролетающие. . .

@u235 5516 / 2869 / 571 Регистрация: 07.11.2019 Сообщений: 4,760
	13.05.2023, 20:05
	В numpy есть функция numpy.lib.stride_tricks.sliding_window_v iew. Попробуйте использовать ее. 0

@u235 5516 / 2869 / 571 Регистрация: 07.11.2019 Сообщений: 4,760
	14.05.2023, 16:45
	MConst, у вас у картинки исходной какой shape? 0

@MConst 2 / 2 / 1 Регистрация: 16.04.2022 Сообщений: 1,032
	14.05.2023, 19:33 [ТС]
	(311, 653, 3) 0

@MConst 2 / 2 / 1 Регистрация: 16.04.2022 Сообщений: 1,032
	14.05.2023, 22:34 [ТС]
	Супер! Спасибо! Только вот не понятно, почему эта функция еще одну ось добавляет. 0

@MConst 2 / 2 / 1 Регистрация: 16.04.2022 Сообщений: 1,032
	14.05.2023, 22:42 [ТС]
	Понятно 0