优点:可以用2080Ti训练;
2. 数据集配置
2.1 类别信息
类别ID | 英文类别名 | 中文类别名 |
---|
100000001 | ship | 船 |
100000002 | aircraft_carrier | 航母 |
100000003 | warcraft | 军舰 |
100000004 | merchant_ship | 商船 |
100000005 | nimitz_carrier | 尼米兹级航母 |
100000006 | enterprise_carrier | 企业级航母 |
100000007 | arleigh_burke_destroyer | 阿利伯克级驱逐舰 |
100000008 | whidbey_island_landing_craft | 惠德贝岛级船坞登陆舰 |
100000009 | perry_frigate | 佩里级护卫舰 |
100000010 | sanantonio_amphibious | 圣安东尼奥级两栖船坞运输舰 |
100000011 | ticonderoga_cruiser | 提康德罗加级巡洋舰 |
100000012 | kitty_hawk_carrier | 小鹰级航母 |
100000013 | merchant_ship | 商船 |
100000014 | container_ship | 集装箱船 |
100000015 | oil_tanker | 油轮 |
100000016 | fishing_boat | 渔船 |
100000017 | yacht | 游艇 |
100000018 | tugboat | 拖船 |
100000019 | cargo_ship | 货船 |
100000020 | passenger_ship | 客船 |
100000021 | warship | 军舰 |
100000022 | submarine | 潜艇 |
100000023 | patrol_boat | 巡逻艇 |
100000024 | minesweeper | 扫雷舰 |
100000025 | supply_ship | 补给舰 |
100000026 | hospital_ship | 医院船 |
100000027 | icebreaker | 破冰船 |
100000028 | research_vessel | 科考船 |
100000029 | lifeboat | 救生船 |
100000030 | barge | 驳船 |
3. 训练模型
python -m visdom.server
python train.py --dataroot ./datasets/sarships \
--name sarships_cyclegan --model cycle_gan