1 В избранное 0 Ответвления 0

OSCHINA-MIRROR/mirrors-Syn2Real

Присоединиться к Gitlife
Откройте для себя и примите участие в публичных проектах с открытым исходным кодом с участием более 10 миллионов разработчиков. Приватные репозитории также полностью бесплатны :)
Присоединиться бесплатно
Клонировать/Скачать
Внести вклад в разработку кода
Синхронизировать код
Отмена
Подсказка: Поскольку Git не поддерживает пустые директории, создание директории приведёт к созданию пустого файла .keep.
Loading...
README.md

Syn2Real

Передача обучения для очистки изображений от дождевых следов с использованием гауссовых процессов

Rajeev Yasarla*, Vishwanath A. Sindagi*, Vishal M. Patel

Ссылка на статью (CVPR '20)

Ссылка на видео

@InProceedings{Yasarla_2020_CVPR,
author = {Yasarla, Rajeev and Sindagi, Vishwanath A. and Patel, Vishal M.},
title = {Syn2Real Transfer Learning for Image Deraining Using Gaussian Processes},
booktitle = {The IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2020}
}Мы предлагаем фреймворк полунадзорного обучения на основе гауссовских процессов, который позволяет сети обучаться очистке изображений от дождевых следов, используя синтетический набор данных, при этом лучше обобщаясь на ненаклассифицированных реальных изображениях. Через широкие эксперименты и аблацию на нескольких сложных наборах данных (например, Rain800, Rain200H и DDN-SIRR), мы показываем, что предложенный метод, когда он обучается на ограниченном количестве меткированных данных, достигает аналогичной производительности по сравнению с полностью меткарованным обучением. Кроме того, мы демонстрируем, что использование ненаклассифицированных реальных изображений в предложенном фреймворке на основе гауссовских процессов приводит к лучшей производительности по сравнению с существующими методами.## Расширение журнала:

Полунадзорное удаление дождевых следов изображений с использованием гауссовских процессов

Ссылка на статью

Предварительные требования:

  1. Linux
  2. Python 2 или 3
  3. Pytorch версии >=1.9
  4. Процессор или NVIDIA GPU + CUDA CuDNN (CUDA 10.2)

Структура набора данных

  1. Скачайте наборы данных с дождями и расположите дождевые изображения и чистые изображения в следующем порядке.
  2. Сохраните названия изображений в текстовый файл (dataset_filename.txt).``` . ├── data | ├── train # Training
    | | ├── derain
    | | | ├── <название_набора> | | | | ├── rain # images with rain | | | | └── norain # clean images | | | └── dataset_filename.txt | └── test # Testing | | ├── derain
    | | | ├── <название_набора> | | | | ├── rain # images with rain | | | | └── norain # clean images | | | └── dataset_filename.txt

## Для тестирования Syn2Real:
 1.  Укажите файл с тестовым набором данных в строке 57 файла `test.py`, например
 val_filename = 'SIRR_test.txt'
2.  Выполните следующую команду
 python test.py -category derain -exp_name DDN_SIRR_withGP
## Для обучения Syn2Real:
1.  Укажите отмеченные, незамеченные и проверочные наборы данных в строках 119-121 файла `train.py`, например
 labeled_name = 'DDN_100_split1.txt'
 unlabeled_name = 'real_input_split1.txt'
 val_filename = 'SIRR_test.txt'
2.  Выполните следующую команду для обучения базовой сети без гауссовских процессов
 python train.py -train_batch_size 2 -category derain -exp_name DDN_SIRR_withoutGP -lambda_GP 0.00 -epoch_start 0
3.  Выполните следующую команду для обучения модели Syn2Real (CVPR'20)
 python train.py -train_batch_size 2 -category derain -exp_name DDN_SIRR_withGP -lambda_GP 0.0015 -epoch_start 0 -version version1
4.  Выполните следующую команду для обучения модели Syn2Real++ (представленной в журнале, модификация ГП на уровне карт признаков)
 python train.py -train_batch_size 2 -category derain -exp_name DDN_SIRR_withGP -lambda_GP 0.0015 -epoch_start 0 -version version2
## Кросс-доменные эксперименты и гауссовские ядра

Кросс-доменные эксперименты выполняются с использованием набора данных DIDMDN как исходного набора данных, а также других наборов данных, таких как Rain800, JORDER_200L, DDN.```
----------------------------------------------------
Исходные наборы данных | Целевые наборы данных                
----------------------------------------------------
DIDMDN                 | Rain800, JORDER_200L, DDN            
----------------------------------------------------

Гауссовские процессы можно моделировать с помощью различных ядер, таких как линейное, квадратичное экспоненциальное или рациональное квадратическое. Обновленный код предоставляет возможность выбора типа ядра

-kernel_type <линейное или квадратичное экспоненциальное или рациональное квадратическое>

Быстрая версия ГП

используйте файл GP_new_fast.py для быстрой версии ГП.

Для использования этого GP_new_fast.py:
    закомментируйте строку 14 в файле `train.py`
    и раскомментируйте строку 15 в файле `train.py`

Кроме того, вы можете использовать файл "train_new_comb.py" вместо "train.py".

В файле "train_new_comb.py" происходит итерационное обучение сети, то есть каждая итерация содержит один шаг обучения с помеченными данными и один шаг обучения с непомеченными данными. Запустите следующую команду для обучения модели Syn2Real (CVPR'20) с использованием файла "train_new_comb.py".

    python train_new_comb.py -train_batch_size 2 -category derain -exp_name DDN_SIRR_withGP -lambda_GP 0.0015 -epoch_start 0 -version version1

Комментарии ( 0 )

Вы можете оставить комментарий после Вход в систему

Введение

В университете Хопкинса используется Syn2Real с открытым исходным кодом для применения алгоритма удаления дождя из изображений с использованием гауссовского процесса. Развернуть Свернуть
MIT
Отмена

Обновления

Пока нет обновлений

Участники

все

Недавние действия

Загрузить больше
Больше нет результатов для загрузки
1
https://api.gitlife.ru/oschina-mirror/mirrors-Syn2Real.git
git@api.gitlife.ru:oschina-mirror/mirrors-Syn2Real.git
oschina-mirror
mirrors-Syn2Real
mirrors-Syn2Real
master