CycleGAN学习笔记

1. 推荐repo: pytorch-CycleGAN-and-pix2pix

优点:可以用2080Ti训练;

2. 数据集配置

SensorDataset
OpticalHRSC2016 [kaggle]

2.1 类别信息

类别ID英文类别名中文类别名
100000001ship
100000002aircraft_carrier航母
100000003warcraft军舰
100000004merchant_ship商船
100000005nimitz_carrier尼米兹级航母
100000006enterprise_carrier企业级航母
100000007arleigh_burke_destroyer阿利伯克级驱逐舰
100000008whidbey_island_landing_craft惠德贝岛级船坞登陆舰
100000009perry_frigate佩里级护卫舰
100000010sanantonio_amphibious圣安东尼奥级两栖船坞运输舰
100000011ticonderoga_cruiser提康德罗加级巡洋舰
100000012kitty_hawk_carrier小鹰级航母
100000013merchant_ship商船
100000014container_ship集装箱船
100000015oil_tanker油轮
100000016fishing_boat渔船
100000017yacht游艇
100000018tugboat拖船
100000019cargo_ship货船
100000020passenger_ship客船
100000021warship军舰
100000022submarine潜艇
100000023patrol_boat巡逻艇
100000024minesweeper扫雷舰
100000025supply_ship补给舰
100000026hospital_ship医院船
100000027icebreaker破冰船
100000028research_vessel科考船
100000029lifeboat救生船
100000030barge驳船

3. 训练模型

python -m visdom.server
python train.py --dataroot ./datasets/sarships \
--name sarships_cyclegan --model cycle_gan
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值