Как да преоразмерите изображение до конкретен размер в PyTorch?

PyTorch е добре позната рамка за дълбоко обучение, която предлага различни инструменти и модули за работа с множество изображения. Той също така предоставя модул „torchvision.transforms“, който има набор от класове и функции за извършване на различни трансформации на изображения, като изрязване, преоразмеряване, завъртане, обръщане, мащабиране и много други. Потребителите могат да използват „ Преоразмеряване() ” и задайте размера, т.е. височината и ширината, за да преоразмерите всяко желано изображение. Този метод връща ново преоразмерено изображение с посочения размер.

Този блог ще демонстрира метода за преоразмеряване на изображение до определен размер в PyTorch.

Как да преоразмерите изображение до конкретен размер в PyTorch?

За да промените размерите на изображение и да го преоразмерите до определен размер в PyTorch, следвайте стъпките по-долу:

• Качете изображение в Google Colab
• Импортирайте необходимите библиотеки
• Прочетете въведеното изображение
• Изчислете и отпечатайте входния размер на изображението
• Създайте трансформация
• Приложете трансформация върху посоченото изображение
• Покажете преоразмереното изображение и неговия размер

Стъпка 1: Качете изображение в Google Colab

Първо отворете Google Colab и щракнете върху маркираните по-долу икони. След това изберете конкретното изображение от компютъра и го качете:

Впоследствие изображението ще бъде качено в Google Colab:

Тук сме качили следното изображение и ще го преоразмерим до определен размер:

Стъпка 2: Импортирайте необходимата библиотека

След това импортирайте необходимите библиотеки. Например импортирахме следните библиотеки:

Тук:

• “ факел за внос ” импортира библиотеката на PyTorch.
• “ импортиране на torchvision.transforms като трансформации ” импортира трансформиращия модул от torchvision, който се използва за предварителна обработка на данни за изображения, преди да ги подаде в невронна мрежа.
• “ от PIL импортиране на изображение ” отваря и записва различни файлови формати на изображения.
• “ импортирайте matplotlib.pyplot като plt ” импортира модула „pyplot”, който се използва за създаване на визуализации и графики:

Стъпка 3: Прочетете входното изображение

След това прочетете въведеното изображение от компютъра. Тук четем „ galaxy_img.jpg ” и съхраняването му в „ input_img ” променлива:

Стъпка 4: Изчислете и отпечатайте входния размер на изображението

След това изчислете размера на входното изображение, т.е. ширина и височина, и го отпечатайте:

Може да се види, че ширината на входното изображение е „384“, а височината на изображението е „576“:

Стъпка 5: Създайте трансформация

След това дефинирайте трансформация, за да промените размера на входното изображение. Потребителите трябва да посочат новия размер, т.е. височина и ширина за новото изображение. Тук сме посочили височина „200“ и ширина „400“:

Стъпка 6: Приложете трансформацията върху входното изображение

Сега приложете горната трансформация върху указаното входно изображение, за да промените размерите му:

Стъпка 7: Показване на преоразмерено изображение и неговия размер

Накрая вижте преоразмереното изображение, като го покажете и проверете размера му:

Изходът по-долу показва, че входното изображение е преоразмерено успешно. Сега ширината му е „400“, а височината му е „200“:

По подобен начин потребителите могат да зададат всеки друг размер, за да преоразмерят изображението. Тук ще преоразмерим същото входно изображение с друг размер, т.е. височина „250“ и ширина „150“:

Това ще преоразмери изображението до нови размери:

Сравнение

Сравнението между оригиналното изображение и преоразмерените изображения с различни размери може да се види по-долу:

Забележка : Можете да получите достъп до нашия бележник Google Colab тук връзка .

Обяснихме ефективно метода за преоразмеряване на изображение до желания размер в PyTorch.

Заключение

За да преоразмерите желаното изображение до определен размер в PyTorch, първо качете желаното изображение в Google Colab. След това импортирайте необходимите библиотеки и прочетете въведеното изображение. След това изчислете и отпечатайте размера на входното изображение. След това използвайте „ Преоразмеряване() ”, за да дефинирате и приложите трансформацията върху желаното входно изображение. Накрая покажете новото преоразмерено изображение и неговия размер. Този блог демонстрира метода за преоразмеряване на желаното изображение до определен размер в PyTorch.