File tree Expand file tree Collapse file tree 1 file changed +12
-6
lines changed Expand file tree Collapse file tree 1 file changed +12
-6
lines changed Original file line number Diff line number Diff line change @@ -32,12 +32,18 @@ class RandomCropOp : public framework::OperatorWithKernel {
32
32
class RandomCropOpMaker : public framework ::OpProtoAndCheckerMaker {
33
33
public:
34
34
void Make () override {
35
- AddInput (" X" , " " );
36
- AddOutput (" Out" , " " );
37
- AddInput (" Seed" , " " );
38
- AddOutput (" SeedOut" , " " ).AsDispensable ();
39
- AddAttr<std::vector<int >>(" shape" , " " );
40
- AddComment (" " );
35
+ AddInput (" X" , " A batch of instances to random crop." );
36
+ AddInput (" Seed" , " The random seed." );
37
+ AddOutput (" Out" , " The cropped instance batch." );
38
+ AddOutput (" SeedOut" , " The random seed after random cropping." )
39
+ .AsDispensable ();
40
+ AddAttr<std::vector<int >>(" shape" , " The shape of a cropped instance." );
41
+ AddComment (R"DOC(
42
+ This operator takes a batch of instance, and do random cropping on each instance.
43
+ It means that cropping positions differs on each instance, which is determined
44
+ by an uniform random generator. All cropped instances have the same shape, which
45
+ is determined by the operator's attribute 'shape'.
46
+ )DOC" );
41
47
}
42
48
};
43
49
You can’t perform that action at this time.
0 commit comments