close
The Wayback Machine - https://web.archive.org/web/20201111224946/https://github.com/PaddlePaddle/PaddleX/pull/117
Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add yolo with iou loss #117

Open
wants to merge 11 commits into
base: develop
from
Open

Conversation

@SunAhong1993
Copy link
Contributor

@SunAhong1993 SunAhong1993 commented May 28, 2020

No description provided.

00
@SunAhong1993 SunAhong1993 changed the title Syf0528 add yolo with iou loss May 28, 2020
new_save_dir = paddlex.pretrain_dir
if backbone == 'ResNet50_vd':
backbone = 'ResNet50_vd_dcn_db_obj365'
assert backbone in obj365_pretrain, "There is not Object365 pretrain weights for {}, you may try ImageNet.".format(

This comment has been minimized.

@jiangjiajun

jiangjiajun May 28, 2020
Collaborator

改成 There is no pretrain weight on Object365 dataset for {}, try use pretrain_weights='IMAGENET' 这样吧

backbone)
url = obj365_pretrain[backbone]
fname = osp.split(url)[-1].split('.')[0]
paddlex.utils.download_and_decompress(url, path=new_save_dir)

This comment has been minimized.

@jiangjiajun

jiangjiajun May 28, 2020
Collaborator

这里得改成hub下载

use_iou_aware_loss=False,
iou_aware_factor=0.4,
use_drop_block=False,
use_dcn_v2=False,

This comment has been minimized.

@jiangjiajun

jiangjiajun May 28, 2020
Collaborator

这里列的参数,是对于yolo所有backbone都适用么

use_iou_aware_loss=self.use_iou_aware_loss,
iou_aware_factor=self.iou_aware_factor,
use_drop_block=self.use_drop_block,
batch_size=self.train_batch_size if hasattr(self, 'train_batch_size') else 8)

This comment has been minimized.

@jiangjiajun

jiangjiajun May 28, 2020
Collaborator

这里怎么还传入了batch_size

use_iou_aware_loss=False,
iou_aware_factor=0.4,
use_drop_block=False,
use_dcn_v2=False,
train_random_shapes=[

This comment has been minimized.

@jiangjiajun

jiangjiajun May 28, 2020
Collaborator

train_random_shapes='Deprecated'

This comment has been minimized.

@jiangjiajun

jiangjiajun May 28, 2020
Collaborator

if train_random_shapes != "Deprecated":
raise Exception(

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Linked issues

Successfully merging this pull request may close these issues.

None yet

2 participants
You can’t perform that action at this time.